seetaface6使用记录
根据landmark,标记出人脸左右眼,和嘴巴, 环境windows7 64位。
根据测试,左眼的点为:36至41,最左边位36,上面37,38,右边39,下边41,40
右眼,最左边42,上面43,44,右边45,下边47,46
嘴巴,最左边48,右边54,上边 49至53,下边,55至58.
#include <iostream>
#include <opencv2/opencv.hpp>
#include "seeta/FaceDetector.h"
#include "seeta/FaceLandmarker.h"
#include "seeta/FaceRecognizer.h"
void extract_feature(const seeta::FaceDetector &FD,
const seeta::FaceLandmarker &FL,
const seeta::FaceRecognizer &FR,
std::string image_path,
float* feature)
{
cv::Mat img = cv::imread(image_path);
SeetaImageData simg;
simg.height = img.rows;
simg.width = img.cols;
simg.channels = img.channels();
simg.data = img.data;
auto faces = FD.detect(simg);
if (faces.size <= 0)
{
std::cout << "no face detected in" << image_path << std::endl;
return;
}
SeetaPointF points[5];
FL.mark(simg, faces.data[0].pos, points);
FR.Extract(simg, points, feature);
}
typedef std::vector<std::vector<SeetaPointF>> doubleSeetaFace;
int extract_landmark(const seeta::FaceDetector &FD, const seeta::FaceLandmarker &FL, const cv::Mat& img, doubleSeetaFace& points)
{
points.clear();
SeetaImageData simg;
simg.height = img.rows;
simg.width = img.cols;
simg.channels = img.channels();
simg.data = img.data;
auto faces = FD.detect(simg);
//size_t face_s = faces.size()
if (faces.size <= 0)
{
std::cout << "no face detected in" << std::endl;
return 0;
}
for (int i = 0; i < faces.size; i++){
SeetaPointF spoints[68];
FL.mark(simg, faces.data[i].pos, spoints);
std::vector<SeetaPointF> pts(spoints, spoints + 68);
points.push_back(pts);
}
return faces.size;
}
int smain()
{
seeta::ModelSetting fd_setting;
fd_setting.set_device(SEETA_DEVICE_CPU);
fd_setting.append("models/face_detector.csta");
seeta::FaceDetector FD(fd_setting);
seeta::ModelSetting fl_setting;
fl_setting.set_device(SEETA_DEVICE_CPU);
fl_setting.append("models/face_landmarker_pts5.csta");
seeta::FaceLandmarker FL(fl_setting);
seeta::ModelSetting fr_setting;
fr_setting.set_id(0);
fr_setting.append("models/face_recognizer.csta");
fr_setting.set_device(SEETA_DEVICE_CPU);
seeta::FaceRecognizer FR(fr_setting);
std::shared_ptr<float> feature1(new float[FR.GetExtractFeatureSize()]);
std::shared_ptr<float> feature2(new float[FR.GetExtractFeatureSize()]);
extract_feature(FD, FL, FR, "1.jpg", feature1.get());
extract_feature(FD, FL, FR, "2.jpg", feature2.get());
float sim = FR.CalculateSimilarity(feature1.get(), feature2.get());
std::cout << "face's similarity in 1.png and 2.png is" << sim << std::endl;
system("pause");
return 0;
}
#include <stdio.h>
void detect_Face()
{
seeta::ModelSetting fd_setting;
fd_setting.set_device(SEETA_DEVICE_CPU);
fd_setting.append("models/face_detector.csta");
seeta::FaceDetector FD(fd_setting);
seeta::ModelSetting fl_setting;
fl_setting.set_device(SEETA_DEVICE_CPU);
fl_setting.append("models/face_landmarker_pts68.csta");
seeta::FaceLandmarker FL(fl_setting);
cv::VideoCapture cap(0);
cv::Mat frame;
doubleSeetaFace points;
int radiusCircle = 2;
cv::Scalar colorCircle1(0, 0, 255); // (B, G, R)
int thicknessCircle1 = 2;
int font_face = cv::FONT_HERSHEY_COMPLEX;
double font_scale = 0.3;
int thickness = 1;
while (cap.read(frame))
{
cv::flip(frame, frame, 1);
int face_nums = extract_landmark(FD, FL, frame, points);
cv::Rect leftEye, rightEye, mouth;
for (auto iter = points.begin(); iter != points.end(); iter++)
{
auto jter = iter->begin();
leftEye.x = (jter+36)->x;
leftEye.y = (jter+37)->y;
leftEye.width = (jter+39)->x - leftEye.x;
leftEye.height = (jter+41)->y - leftEye.y;
leftEye.x -= 10;
leftEye.y -= 10;
leftEye.width += 20;
leftEye.height += 20;
rightEye.x = (jter+42)->x;
rightEye.y = (jter+43)->y;
rightEye.width = (jter+45)->x - rightEye.x;
rightEye.height = (jter+47)->y - rightEye.y;
rightEye.x -= 10;
rightEye.y -= 10;
rightEye.width += 20;
rightEye.height += 20;
mouth.x = (jter+48)->x;
mouth.y = (jter+50)->y;
mouth.width = (jter+54)->x - mouth.x;
mouth.height = (jter+57)->y - mouth.y;
mouth.x -= 10;
mouth.y -= 10;
mouth.width += 20;
mouth.height += 20;
cv::rectangle(frame, leftEye, colorCircle1);
cv::rectangle(frame, rightEye, colorCircle1);
cv::rectangle(frame, mouth, colorCircle1);
#if 0
int count = 0;
std::vector<SeetaPointF> pt(iter->begin(), iter->end());
for (auto jter = iter->begin(); jter != iter->end(); jter++)
{
cv::Point ps(jter->x, jter->y);
cv::Point ws(ps.x, ps.y - 5);
/*char str[256];
sprintf_s(str, 256, "%d", count++);
cv::putText(frame, str, ws, font_face, font_scale, cv::Scalar(0, 255, 255), thickness, 8, 0);
cv::circle(frame, ps, radiusCircle, colorCircle1, thicknessCircle1);*/
}
#endif
}
//cv::resize(frame, frame, cv::Size(frame.cols * 2.5, frame.rows * 2.5));
cv::imshow("detect.jpg", frame);
cv::waitKey(20);
}
cap.release();
}
int main(int argc, char** argv)
{
detect_Face();
}
每天快乐敲代码,快乐生活