1 #include <stdio.h> 2 #include <iostream> 3 #include "opencv2/core/core.hpp" 4 #include "opencv2/features2d/features2d.hpp" 5 #include "opencv2/highgui/highgui.hpp" 6 7 using namespace cv; 8 9 void readme(); 10 11 /** @function main */ 12 int main( int argc, char** argv ) 13 { 14 if( argc != 3 ) 15 { return -1; } 16 17 Mat img_1 = imread( argv[1], CV_LOAD_IMAGE_GRAYSCALE ); 18 Mat img_2 = imread( argv[2], CV_LOAD_IMAGE_GRAYSCALE ); 19 20 if( !img_1.data || !img_2.data ) 21 { return -1; } 22 23 //-- Step 1: Detect the keypoints using SURF Detector 24 int minHessian = 400; 25 26 SurfFeatureDetector detector( minHessian ); 27 28 std::vector<KeyPoint> keypoints_1, keypoints_2; 29 30 detector.detect( img_1, keypoints_1 ); 31 detector.detect( img_2, keypoints_2 ); 32 33 //-- Step 2: Calculate descriptors (feature vectors) 34 SurfDescriptorExtractor extractor; 35 36 Mat descriptors_1, descriptors_2; 37 38 extractor.compute( img_1, keypoints_1, descriptors_1 ); 39 extractor.compute( img_2, keypoints_2, descriptors_2 ); 40 41 //-- Step 3: Matching descriptor vectors with a brute force matcher 42 BruteForceMatcher< L2<float> > matcher; 43 std::vector< DMatch > matches; 44 matcher.match( descriptors_1, descriptors_2, matches ); 45 46 //-- Draw matches 47 Mat img_matches; 48 drawMatches( img_1, keypoints_1, img_2, keypoints_2, matches, img_matches ); 49 50 //-- Show detected matches 51 imshow("Matches", img_matches ); 52 53 waitKey(0); 54 55 return 0; 56 } 57 58 /** @function readme */ 59 void readme() 60 { std::cout << " Usage: ./SURF_descriptor <img1> <img2>" << std::endl; }
