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 { readme(); 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 { std::cout<< " --(!) Error reading images " << std::endl; 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 //-- Draw keypoints 34 Mat img_keypoints_1; Mat img_keypoints_2; 35 36 drawKeypoints( img_1, keypoints_1, img_keypoints_1, Scalar::all(-1), DrawMatchesFlags::DEFAULT ); 37 drawKeypoints( img_2, keypoints_2, img_keypoints_2, Scalar::all(-1), DrawMatchesFlags::DEFAULT ); 38 39 //-- Show detected (drawn) keypoints 40 imshow("Keypoints 1", img_keypoints_1 ); 41 imshow("Keypoints 2", img_keypoints_2 ); 42 43 waitKey(0); 44 45 return 0; 46 } 47 48 /** @function readme */ 49 void readme() 50 { std::cout << " Usage: ./SURF_detector <img1> <img2>" << std::endl; }
