|
|
@ -6,13 +6,16 @@ |
|
|
|
|
|
|
|
|
|
|
|
#include "opencv2/imgproc.hpp" |
|
|
|
#include "opencv2/imgproc.hpp" |
|
|
|
#include "opencv2/core.hpp" |
|
|
|
#include "opencv2/core.hpp" |
|
|
|
|
|
|
|
#ifdef HAVE_OPENCV_DNN |
|
|
|
#include "opencv2/dnn.hpp" |
|
|
|
#include "opencv2/dnn.hpp" |
|
|
|
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
|
|
#include <algorithm> |
|
|
|
#include <algorithm> |
|
|
|
|
|
|
|
|
|
|
|
namespace cv |
|
|
|
namespace cv |
|
|
|
{ |
|
|
|
{ |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef HAVE_OPENCV_DNN |
|
|
|
class FaceDetectorYNImpl : public FaceDetectorYN |
|
|
|
class FaceDetectorYNImpl : public FaceDetectorYN |
|
|
|
{ |
|
|
|
{ |
|
|
|
public: |
|
|
|
public: |
|
|
@ -273,6 +276,7 @@ private: |
|
|
|
|
|
|
|
|
|
|
|
std::vector<Rect2f> priors; |
|
|
|
std::vector<Rect2f> priors; |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
|
|
Ptr<FaceDetectorYN> FaceDetectorYN::create(const String& model, |
|
|
|
Ptr<FaceDetectorYN> FaceDetectorYN::create(const String& model, |
|
|
|
const String& config, |
|
|
|
const String& config, |
|
|
@ -283,7 +287,12 @@ Ptr<FaceDetectorYN> FaceDetectorYN::create(const String& model, |
|
|
|
const int backend_id, |
|
|
|
const int backend_id, |
|
|
|
const int target_id) |
|
|
|
const int target_id) |
|
|
|
{ |
|
|
|
{ |
|
|
|
|
|
|
|
#ifdef HAVE_OPENCV_DNN |
|
|
|
return makePtr<FaceDetectorYNImpl>(model, config, input_size, score_threshold, nms_threshold, top_k, backend_id, target_id); |
|
|
|
return makePtr<FaceDetectorYNImpl>(model, config, input_size, score_threshold, nms_threshold, top_k, backend_id, target_id); |
|
|
|
|
|
|
|
#else |
|
|
|
|
|
|
|
CV_UNUSED(model); CV_UNUSED(config); CV_UNUSED(input_size); CV_UNUSED(score_threshold); CV_UNUSED(nms_threshold); CV_UNUSED(top_k); CV_UNUSED(backend_id); CV_UNUSED(target_id); |
|
|
|
|
|
|
|
CV_Error(cv::Error::StsNotImplemented, "cv::FaceDetectorYN requires enabled 'dnn' module."); |
|
|
|
|
|
|
|
#endif |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
} // namespace cv
|
|
|
|
} // namespace cv
|
|
|
|