1 #include "opencv2/highgui/highgui.hpp" 2 #include "opencv2/imgproc/imgproc.hpp" 3 #include <iostream> 4 #include <stdio.h> 5 6 using namespace cv; 7 8 /** @function main */ 9 int main(int argc, char** argv) 10 { 11 Mat src, src_gray; 12 13 /// Read the image 14 src = imread( argv[1], 1 ); 15 16 if( !src.data ) 17 { return -1; } 18 19 /// Convert it to gray 20 cvtColor( src, src_gray, CV_BGR2GRAY ); 21 22 /// Reduce the noise so we avoid false circle detection 23 GaussianBlur( src_gray, src_gray, Size(9, 9), 2, 2 ); 24 25 vector<Vec3f> circles; 26 27 /// Apply the Hough Transform to find the circles 28 HoughCircles( src_gray, circles, CV_HOUGH_GRADIENT, 1, src_gray.rows/8, 200, 100, 0, 0 ); 29 30 /// Draw the circles detected 31 for( size_t i = 0; i < circles.size(); i++ ) 32 { 33 Point center(cvRound(circles[i][0]), cvRound(circles[i][1])); 34 int radius = cvRound(circles[i][2]); 35 // circle center 36 circle( src, center, 3, Scalar(0,255,0), -1, 8, 0 ); 37 // circle outline 38 circle( src, center, radius, Scalar(0,0,255), 3, 8, 0 ); 39 } 40 41 /// Show your results 42 namedWindow( "Hough Circle Transform Demo", CV_WINDOW_AUTOSIZE ); 43 imshow( "Hough Circle Transform Demo", src ); 44 45 waitKey(0); 46 return 0; 47 }
