OpenCV Shi-Tomasi角点检测子

 1 #include "opencv2/highgui/highgui.hpp"
 2 #include "opencv2/imgproc/imgproc.hpp"
 3 #include <iostream>
 4 #include <stdio.h>
 5 #include <stdlib.h>
 6 
 7 using namespace cv;
 8 using namespace std;
 9 
10 /// Global variables
11 Mat src, src_gray;
12 
13 int maxCorners = 23;
14 int maxTrackbar = 100;
15 
16 RNG rng(12345);
17 char* source_window = "Image";
18 
19 /// Function header
20 void goodFeaturesToTrack_Demo( int, void* );
21 
22 /**
23  * @function main
24  */
25 int main( int argc, char** argv )
26 {
27   /// Load source image and convert it to gray
28   src = imread( argv[1], 1 );
29   cvtColor( src, src_gray, CV_BGR2GRAY );
30 
31   /// Create Window
32   namedWindow( source_window, CV_WINDOW_AUTOSIZE );
33 
34   /// Create Trackbar to set the number of corners
35   createTrackbar( "Max  corners:", source_window, &maxCorners, maxTrackbar, goodFeaturesToTrack_Demo );
36 
37   imshow( source_window, src );
38 
39   goodFeaturesToTrack_Demo( 0, 0 );
40 
41   waitKey(0);
42   return(0);
43 }
44 
45 /**
46  * @function goodFeaturesToTrack_Demo.cpp
47  * @brief Apply Shi-Tomasi corner detector
48  */
49 void goodFeaturesToTrack_Demo( int, void* )
50 {
51   if( maxCorners < 1 ) { maxCorners = 1; }
52 
53   /// Parameters for Shi-Tomasi algorithm
54   vector<Point2f> corners;
55   double qualityLevel = 0.01;
56   double minDistance = 10;
57   int blockSize = 3;
58   bool useHarrisDetector = false;
59   double k = 0.04;
60 
61   /// Copy the source image
62   Mat copy;
63   copy = src.clone();
64 
65   /// Apply corner detection
66   goodFeaturesToTrack( src_gray,
67                corners,
68                maxCorners,
69                qualityLevel,
70                minDistance,
71                Mat(),
72                blockSize,
73                useHarrisDetector,
74                k );
75 
76 
77   /// Draw corners detected
78   cout<<"** Number of corners detected: "<<corners.size()<<endl;
79   int r = 4;
80   for( int i = 0; i < corners.size(); i++ )
81      { circle( copy, corners[i], r, Scalar(rng.uniform(0,255), rng.uniform(0,255),
82               rng.uniform(0,255)), -1, 8, 0 ); }
83 
84   /// Show what you got
85   namedWindow( source_window, CV_WINDOW_AUTOSIZE );
86   imshow( source_window, copy );
87 }