face.hpp

// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.
 
#ifndef OPENCV_OBJDETECT_FACE_HPP
#define OPENCV_OBJDETECT_FACE_HPP
 
#include <opencv2/core.hpp>
 
namespace cv
{
 
 
class CV_EXPORTS_W FaceDetectorYN
{
public:
    virtual ~FaceDetectorYN() {}
 
    CV_WRAP virtual void setInputSize(const Size& input_size) = 0;
 
    CV_WRAP virtual Size getInputSize() = 0;
 
    CV_WRAP virtual void setScoreThreshold(float score_threshold) = 0;
 
    CV_WRAP virtual float getScoreThreshold() = 0;
 
    CV_WRAP virtual void setNMSThreshold(float nms_threshold) = 0;
 
    CV_WRAP virtual float getNMSThreshold() = 0;
 
    CV_WRAP virtual void setTopK(int top_k) = 0;
 
    CV_WRAP virtual int getTopK() = 0;
 
    CV_WRAP virtual int detect(InputArray image, OutputArray faces) = 0;
 
    CV_WRAP static Ptr<FaceDetectorYN> create(CV_WRAP_FILE_PATH const String& model,
                                              CV_WRAP_FILE_PATH const String& config,
                                              const Size& input_size,
                                              float score_threshold = 0.9f,
                                              float nms_threshold = 0.3f,
                                              int top_k = 5000,
                                              int backend_id = 0,
                                              int target_id = 0);
 
    CV_WRAP static Ptr<FaceDetectorYN> create(const String& framework,
                                              const std::vector<uchar>& bufferModel,
                                              const std::vector<uchar>& bufferConfig,
                                              const Size& input_size,
                                              float score_threshold = 0.9f,
                                              float nms_threshold = 0.3f,
                                              int top_k = 5000,
                                              int backend_id = 0,
                                              int target_id = 0);
 
};
 
class CV_EXPORTS_W FaceRecognizerSF
{
public:
    virtual ~FaceRecognizerSF() {}
 
    enum DisType { FR_COSINE=0, FR_NORM_L2=1 };
 
    CV_WRAP virtual void alignCrop(InputArray src_img, InputArray face_box, OutputArray aligned_img) const = 0;
 
    CV_WRAP virtual void feature(InputArray aligned_img, OutputArray face_feature) = 0;
 
    CV_WRAP virtual double match(InputArray face_feature1, InputArray face_feature2, int dis_type = FaceRecognizerSF::FR_COSINE) const = 0;
 
    CV_WRAP static Ptr<FaceRecognizerSF> create(CV_WRAP_FILE_PATH const String& model, CV_WRAP_FILE_PATH const String& config, int backend_id = 0, int target_id = 0);
};
 
} // namespace cv
 
#endif