QFacefusion/faceclassifier.h at main · QQxiaoming/QFacefusion · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# ifndef FACECLASSIFIER
# define FACECLASSIFIER
#include <fstream>
#include <sstream>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
#include <onnxruntime_cxx_api.h>
#include "onnxbase.h"
#include "utils.h"

class FaceClassifier : public OnnxBase
{
public:
	enum FaceGender {
		MALE = 0,
		FEMALE = 1,
	};
	enum FaceAge {
		AGE_0_2 = 0,
		AGE_3_9 = 1,
		AGE_10_19 = 2,
		AGE_20_29 = 3,
		AGE_30_39 = 4,
		AGE_40_49 = 5,
		AGE_50_59 = 6,
		AGE_60_69 = 7,
		AGE_70_100 = 8,
	};
	enum FaceRace {
		WHITE = 0,
		BLACK = 1,
		LATINO = 2,
		ASIAN = 3,
		INDIAN = 4,
		ARABIC = 5,
	};

	FaceClassifier(std::string modelpath, LoggerCallback logger = nullptr, void* user_data = nullptr);
	std::vector<int> detect(cv::Mat srcimg, const std::vector<cv::Point2f> face_landmark_5);
private:
	void preprocess(cv::Mat img, const std::vector<cv::Point2f> face_landmark_5);
	std::vector<float> input_image;
	int input_height;
	int input_width;
    std::vector<cv::Point2f> normed_template;
	const float FAIRFACE_MODEL_MEAN[3] = {0.485, 0.456, 0.406};
	const float FAIRFACE_MODEL_STD[3] = {0.229, 0.224, 0.225};
};
#endif