opencvsharp/src/OpenCvSharpExtern/face_detector_yn.cpp at df8e814e6509f5b4d7750210ebd993f135a4a9bc · shimat/opencvsharp · 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
50
51
52
53
54
#include "face_detector_yn.h"
#include <opencv2/core/cvstd.hpp>
#include <opencv2/core/cvstd_wrapper.hpp>
#include <opencv2/core/mat.hpp>
#include <opencv2/core/types_c.h>
#include <opencv2/objdetect/face.hpp>
#include <opencv2/opencv_modules.hpp>

cv::FaceDetectorYN* cveFaceDetectorYNCreate(
	cv::String* model,
	cv::String* config,
	CvSize* inputSize,
	float scoreThreshold,
	float nmsThreshold,
	int topK,
	int backendId,
	int targetId,
	cv::Ptr<cv::FaceDetectorYN>** sharedPtr)
{
#ifdef HAVE_OPENCV_OBJDETECT
	cv::Ptr<cv::FaceDetectorYN> ptr = cv::FaceDetectorYN::create(
		*model,
		*config,
		*inputSize,
		scoreThreshold,
		nmsThreshold,
		topK,
		backendId,
		targetId);
	*sharedPtr = new cv::Ptr<cv::FaceDetectorYN>(ptr);
	return (*sharedPtr)->get();
#else
	throw_no_objdetect();
#endif
}

void cveFaceDetectorYNRelease(cv::Ptr<cv::FaceDetectorYN>** faceDetector)
{
#ifdef HAVE_OPENCV_OBJDETECT
	delete* faceDetector;
	*faceDetector = 0;
#else
	throw_no_objdetect();
#endif
}

int cveFaceDetectorYNDetect(cv::FaceDetectorYN* faceDetector, cv::_InputArray* image, cv::_OutputArray* faces)
{
#ifdef HAVE_OPENCV_OBJDETECT
	return faceDetector->detect(*image, *faces);
#else
	throw_no_objdetect();
#endif
}