mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-07 17:44:04 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed PyramidAdaptedFeatureDetector (when mask is not empty) (#1206)

Browse Source
This commit is contained in:
Maria Dimashova 2011-07-08 14:41:43 +00:00
parent c380dd579b
commit 370130e464
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
modules/features2d/src/detectors.cpp
View File

@ -602,11 +602,22 @@ bool PyramidAdaptedFeatureDetector::empty() const
void PyramidAdaptedFeatureDetector::detectImpl( const Mat& image, vector<KeyPoint>& keypoints, const Mat& mask ) const void PyramidAdaptedFeatureDetector::detectImpl( const Mat& image, vector<KeyPoint>& keypoints, const Mat& mask ) const
{ {
Mat src = image; Mat src = image;
Mat src_mask = mask;
Mat dilated_mask;
if( !mask.empty() )
{
dilate( mask, dilated_mask, Mat() );
Mat mask255( mask.size(), CV_8UC1, Scalar(0) );
mask255.setTo( Scalar(255), dilated_mask != 0 );
dilated_mask = mask255;
}
for( int l = 0, multiplier = 1; l <= maxLevel; ++l, multiplier *= 2 ) for( int l = 0, multiplier = 1; l <= maxLevel; ++l, multiplier *= 2 )
{ {
// Detect on current level of the pyramid // Detect on current level of the pyramid
vector<KeyPoint> new_pts; vector<KeyPoint> new_pts;
detector->detect( src, new_pts, mask ); detector->detect( src, new_pts, src_mask );
for( vector<KeyPoint>::iterator it = new_pts.begin(), end = new_pts.end(); it != end; ++it) for( vector<KeyPoint>::iterator it = new_pts.begin(), end = new_pts.end(); it != end; ++it)
{ {
it->pt.x *= multiplier; it->pt.x *= multiplier;
@ -620,10 +631,16 @@ void PyramidAdaptedFeatureDetector::detectImpl( const Mat& image, vector<KeyPoin
if( l < maxLevel ) if( l < maxLevel )
{ {
Mat dst; Mat dst;
pyrDown(src, dst); pyrDown( src, dst );
src = dst; src = dst;
if( !mask.empty() )
resize( dilated_mask, src_mask, src.size(), 0, 0, CV_INTER_AREA );
} }
} }
if( !mask.empty() )
KeyPointsFilter::runByPixelsMask( keypoints, mask );
} }
} }