diff --git a/modules/objdetect/include/opencv2/objdetect/objdetect.hpp b/modules/objdetect/include/opencv2/objdetect/objdetect.hpp index a8dedb1d6..3300c2016 100644 --- a/modules/objdetect/include/opencv2/objdetect/objdetect.hpp +++ b/modules/objdetect/include/opencv2/objdetect/objdetect.hpp @@ -478,9 +478,11 @@ public: }; void setMaskGenerator(Ptr maskGenerator); Ptr getMaskGenerator(); + + void setFaceDetectionMaskGenerator(); + protected: Ptr maskGenerator; - Ptr getDefaultMaskGenerator(); }; diff --git a/modules/objdetect/src/cascadedetect.cpp b/modules/objdetect/src/cascadedetect.cpp index 6c90e7def..7f05b0a6e 100644 --- a/modules/objdetect/src/cascadedetect.cpp +++ b/modules/objdetect/src/cascadedetect.cpp @@ -787,13 +787,11 @@ Ptr FeatureEvaluator::create( int featureType ) CascadeClassifier::CascadeClassifier() { - maskGenerator=getDefaultMaskGenerator(); } CascadeClassifier::CascadeClassifier(const string& filename) { load(filename); - maskGenerator=getDefaultMaskGenerator(); } CascadeClassifier::~CascadeClassifier() @@ -872,12 +870,12 @@ Ptr CascadeClassifier::getMaskGenerator() return maskGenerator; } -Ptr CascadeClassifier::getDefaultMaskGenerator() +void CascadeClassifier::setFaceDetectionMaskGenerator() { #ifdef HAVE_TEGRA_OPTIMIZATION - return tegra::getCascadeClassifierMaskGenerator(*this); + setMaskGenerator(tegra::getCascadeClassifierMaskGenerator(*this)); #else - return Ptr(); + setMaskGenerator(Ptr()); #endif }