1 #include <opencv2/core/core.hpp>
2 #include <opencv2/highgui/highgui.hpp>
3 #include <opencv2/imgproc/imgproc.hpp>
4 #include <opencv2/opencv.hpp>
5
6 #include <vector>
7 #include <iostream>
8
9 using namespace cv; 10 using namespace std; 11
12
13 typedef struct Box 14 { 15 double x; 16 double y; 17 double r; 18 }Box; 19
20 Box circleLeastFit(const vector<CvPoint *> &points) 21 { 22
23 Box box; 24 box.x = 0.0f; 25 box.y = 0.0f; 26 box.r = 0.0f; 27
28 if (points.size() < 3) 29 { 30 return box; 31 } 32
33 int i=0; 34
35 double X1=0; 36 double Y1=0; 37 double X2=0; 38 double Y2=0; 39 double X3=0; 40 double Y3=0; 41 double X1Y1=0; 42 double X1Y2=0; 43 double X2Y1=0; 44
45 int Sum = points.size(); 46 for (i=0;i<Sum;i++) 47 { 48 X1 = X1 + points[i]->x; 49 Y1 = Y1 + points[i]->y; 50 X2 = X2 + points[i]->x*points[i]->x; 51 Y2 = Y2 + points[i]->y*points[i]->y; 52 X3 = X3 + points[i]->x*points[i]->x*points[i]->x; 53 Y3 = Y3 + points[i]->y*points[i]->y*points[i]->y; 54 X1Y1 = X1Y1 + points[i]->x*points[i]->y; 55 X1Y2 = X1Y2 + points[i]->x*points[i]->y*points[i]->y; 56 X2Y1 = X2Y1 + points[i]->x*points[i]->x*points[i]->y; 57 } 58
59 double C,D,E,G,H,N; 60 double a,b,c; 61 N = points.size(); 62 C = N*X2 - X1*X1; 63 D = N*X1Y1 - X1*Y1; 64 E = N*X3 + N*X1Y2 - (X2+Y2)*X1; 65 G = N*Y2 - Y1*Y1; 66 H = N*X2Y1 + N*Y3 - (X2+Y2)*Y1; 67 a = (H*D-E*G)/(C*G-D*D); 68 b = (H*C-E*D)/(D*D-G*C); 69 c = -(a*X1 + b*Y1 + X2 + Y2)/N; 70
71 double A,B,R; 72 A = a/(-2); 73 B = b/(-2); 74 R = sqrt(a*a+b*b-4*c)/2; 75
76 box.x= A; 77 box.y = B; 78 box.r = R; 79
80 return box; 81 } 82
83 //获取自定义核
84 void f_GetElement(int **p,IplImage *src)//为什么p一定要是二级指针？？？
85 { 86 int length = (src->width)*(src->height);//图片大小
87 *p = (int*)malloc(sizeof(int)*length);//动态分配内存，大小为(sizeof(int)*length)
88 memset(*p,0,sizeof(int)*length);//初始化内存区间 p为0
89
90 for(int y=0; y<src->height; y++) 91 { 92 uchar *ptr = (uchar*)(src->imageData + y*src->widthStep); 93
94 int b = y*src->widthStep;//???
95 for(int x=0; x<src->width; x++) 96 { 97 if(ptr[x]==255) 98 { 99 (*p)[b+x] = 1; 100 } 101 else
102 { 103 (*p)[b+x] = 0; 104 } 105 } 106 } 107 } 108
109
110 int main() 111 { 112
113 IplImage* src = cvLoadImage("5.bmp",1); 114
115 IplImage *gray_image = cvCreateImage(cvGetSize(src),8,1); 116 IplImage *rgb_image = cvCreateImage(cvGetSize(src),8,3); 117 cvZero(gray_image); 118 cvZero(rgb_image); 119
120 cvCvtColor(src,gray_image,CV_RGB2GRAY); 121
122 cvCvtColor(gray_image,rgb_image,CV_GRAY2BGR); 123
124
125 CvMemStorage *storage = cvCreateMemStorage(0); 126 CvSeq *first_contours = NULL,*temp_contours = NULL; 127
128 int n = cvFindContours(gray_image,storage,&first_contours,sizeof(CvContour), CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE); 129 printf("n = %d\n",n); 130
131 temp_contours = first_contours; 132
133 for (;temp_contours != NULL; temp_contours = temp_contours->h_next) 134 { 135 vector <CvPoint*> pt_vec; 136 CvPoint *pt; 137 //cvDrawContours(rgb_image, temp_contours, CV_RGB(0, 0, 255), CV_RGB(0, 0, 255), 0,2,CV_FILLED, cvPoint(0, 0));
138 CvRect rc =cvBoundingRect(temp_contours,0); 139
140 cvRectangle(rgb_image, cvPoint(rc.x, rc.y), cvPoint(rc.x + rc.width, rc.y + rc.height),CV_RGB(255, 0, 0), 1, 8, 0); 141
142 for (int i = 0; i < temp_contours->total;i++) 143 { 144 pt = (CvPoint *)cvGetSeqElem(temp_contours,i); 145 pt_vec.push_back(pt); 146 } 147
148 double x = 0.0; 149 double y = 0.0; 150 double r = 0.0; 151 vector<Box *> box; 152 Box box1; 153 box1 = circleLeastFit(pt_vec); 154 printf("%lf %lf %lf \n",box1.x,box1.y,box1.r); 155 cvCircle(rgb_image,cvPoint(box1.x,box1.y),box1.r,cvScalar(155,255,0),5,8,0); 156 cvCircle(rgb_image,cvPoint(box1.x,box1.y),5,cvScalar(155,255,155),5,8,0); 157
158 } 159
160 cvNamedWindow("32"); 161 cvShowImage("32",rgb_image); 162 cvWaitKey(0); 163
164 return 0; 165 }
