Made changes in the method of inserting prefilters into the CascadeClassifier class.

This commit is contained in:
Leonid Beynenson 2011-10-12 11:32:01 +00:00
parent 8f108b4e7e
commit da3a60a5a7
2 changed files with 6 additions and 0 deletions

View File

@ -474,6 +474,7 @@ public:
{
public:
virtual cv::Mat generateMask(const cv::Mat& src)=0;
virtual void initializeMask(const cv::Mat& src) {};
};
void setMaskGenerator(Ptr<MaskGenerator> maskGenerator);
Ptr<MaskGenerator> getMaskGenerator();

View File

@ -1029,6 +1029,11 @@ void CascadeClassifier::detectMultiScale( const Mat& image, vector<Rect>& object
objects.clear();
if (!maskGenerator.empty()) {
maskGenerator->initializeMask(image);
}
if( maxObjectSize.height == 0 || maxObjectSize.width == 0 )
maxObjectSize = image.size();