openCV 车牌分割 图像的读入，二值化，分割

#include <cv.h> #include <cxcore.h> #include <highgui.h> #include <iostream> using namespace std; #pragma comment(lib,"cv.lib") #pragma comment(lib,"cxcore.lib") #pragma comment(lib,"highgui.lib") #define T 27 #define T1 2 #define S(image,x,y) ((uchar*)(image->imageData + image->widthStep*(y)))[(x)] //S void main() { IplImage *src; IplImage *pImg8u=NULL; //灰度图 IplImage *pImg8uSmooth=NULL; //高斯滤波后的图 IplImage *pImgCanny=NULL; //二值化的图 IplImage *pImgHist=NULL; //直方图 int hist_size=155; float range_0[]={0,256}; float *ranges[]={range_0}; int i,j,bin_w; float max_value,min_value; int min_dx,max_dx; int row_start,row_end;//用来记录车牌开始，结束行 int col_start,col_end;//用来记录车牌开始，结束列 int count=0;//用来记录行或列的白点个数 src=cvLoadImage("12.bmp",-1); pImg8uSmooth=cvCreateImage(cvGetSize(src),IPL_DEPTH_8U,1); pImg8u=cvCreateImage(cvGetSize(src),IPL_DEPTH_8U,1); pImgCanny=cvCreateImage(cvGetSize(src),IPL_DEPTH_8U,1); cvCvtColor(src,pImg8u,CV_RGB2GRAY); //灰度化 cvSmooth(pImg8u,pImg8uSmooth,CV_GAUSSIAN,3,0,0);//高斯滤波 cvCanny(pImg8uSmooth,pImgCanny,100,200,3); //二值化 /*cvDilate(pImgCanny,pImgCanny,0,1); cvErode(pImgCanny,pImgCanny,0,1);*/ row_start=0; row_end=0; col_start=0; col_end=0; int row[120]; int col[340]; int k; k=0; bool flag=false; for(j=0;j<pImgCanny->height;j++) //找到上行开始 { count=0; for(i=0;i<pImgCanny->width-1;i++) { if(S(pImgCanny,i,j)!=S(pImgCanny,i+1,j)) //统计行跳数 count++; if(count>T) { row[k]=j; k++; break; } } } for(i=0;i<k;i++) //从上边开始，3行连续时认为是起始行 { if((row[i]==row[i+1]-1)&&(row[i+1]==row[i+2]-1)){ row_start=row[i]; break; } } cout<<"the start row:"<<row_start<<endl; cvLine(pImg8u,cvPoint(0,row_start),cvPoint(src->width,row_start),cvScalar(255,0,0),1,8,0); for(i=k-1;i>row_start;i--) //从下边开始，3行连续时认为是起始行 { if((row[i]==row[i-1]+1)&&(row[i-1]==row[i-2]+1)){ row_end=row[i]; break; } } cout<<"the end row:"<<row_end<<endl; cvLine(pImg8u,cvPoint(0,row_end),cvPoint(src->width,row_end),cvScalar(255,0,0),1,8,0); flag=false; for(i=10;i<pImgCanny->width;i++) //找到左列开始 { count=0; for(j=row_start;j<row_end;j++) { if(S(pImgCanny,i,j)==255) count++; if(count>T1) { col_start=i; flag=true; break; } } if(flag) break; } cout<<"the start col:"<<col_start<<endl; cvLine(pImg8u,cvPoint(col_start,row_start),cvPoint(col_start,row_end),cvScalar(255,0,0),1,8,0); flag=false; for(i=pImgCanny->width-10;i>col_start;i--) //找到右列开始 { count=0; for(j=row_start;j<row_end;j++) { if(S(pImgCanny,i,j)==255) count++; if(count>T1) { col_end=i; flag=true; break; } } if(flag) break; } cout<<"the end col:"<<col_end<<endl; cvLine(pImg8u,cvPoint(col_end,row_start),cvPoint(col_end,row_end),cvScalar(255,0,0),1,8,0); CvRect ROI_rect; //获得图片感兴趣区域 ROI_rect.x=col_start; ROI_rect.y=row_start; ROI_rect.width=col_end-col_start; ROI_rect.height=row_end-row_start; IplImage *pImg8uROI=NULL; //感兴趣的图片 cvSetImageROI(pImg8u,ROI_rect); pImg8uROI=cvCreateImage(cvSize(ROI_rect.width,ROI_rect.height),IPL_DEPTH_8U,1); cvCopy(pImg8u,pImg8uROI); cvResetImageROI(pImg8u); int nWidth=409;//(409,90)分别为感兴趣图像的宽度与高度 int nHeight=90; IplImage *pImgResize=NULL; //归一化的灰度图 pImgResize=cvCreateImage(cvSize(nWidth,nHeight),IPL_DEPTH_8U,1); cvResize(pImg8uROI,pImgResize,CV_INTER_LINEAR); //线性插值 int nCharWidth=45; int nSpace=12; for(i=0;i<7;i++) //得到每个字符的双边界 { switch(i){ case 0: case 1: col[i*2]=i*nCharWidth+i*nSpace; col[i*2+1]=(i+1)*nCharWidth+i*nSpace; break; case 2: case 3: case 4: case 5: case 6: col[i*2]=i*nCharWidth+i*nSpace+22; col[i*2+1]=(i+1)*nCharWidth+i*nSpace+22; break; } } for(i=0;i<14;i++) //画出每个字符的区域 { cvLine(pImgResize,cvPoint(col[i],0),cvPoint(col[i],nHeight),cvScalar(255,0,0),1,8,0); //cout<<col[i*2]<<" "<<col[2*i+1]<<" "; } IplImage *pImgCharOne=NULL; IplImage *pImgCharTwo=NULL; IplImage *pImgCharThree=NULL; IplImage *pImgCharFour=NULL; IplImage *pImgCharFive=NULL; IplImage *pImgCharSix=NULL; IplImage *pImgCharSeven=NULL; pImgCharOne=cvCreateImage(cvSize(nCharWidth,nHeight),IPL_DEPTH_8U,1); pImgCharTwo=cvCreateImage(cvSize(nCharWidth,nHeight),IPL_DEPTH_8U,1); pImgCharThree=cvCreateImage(cvSize(nCharWidth,nHeight),IPL_DEPTH_8U,1); pImgCharFour=cvCreateImage(cvSize(nCharWidth,nHeight),IPL_DEPTH_8U,1); pImgCharFive=cvCreateImage(cvSize(nCharWidth,nHeight),IPL_DEPTH_8U,1); pImgCharSix=cvCreateImage(cvSize(nCharWidth,nHeight),IPL_DEPTH_8U,1); pImgCharSeven=cvCreateImage(cvSize(nCharWidth,nHeight),IPL_DEPTH_8U,1); CvRect ROI_rect1; ROI_rect1.x=col[0]; ROI_rect1.y=0; ROI_rect1.width=nCharWidth; ROI_rect1.height=nHeight; cvSetImageROI(pImgResize,ROI_rect1); cvCopy(pImgResize,pImgCharOne,NULL); //获取第1个字符 cvResetImageROI(pImgResize); ROI_rect1.x=col[2]; ROI_rect1.y=0; ROI_rect1.width=nCharWidth; ROI_rect1.height=nHeight; cvSetImageROI(pImgResize,ROI_rect1); cvCopy(pImgResize,pImgCharTwo,NULL); //获取第2个字符 cvResetImageROI(pImgResize); ROI_rect1.x=col[4]; ROI_rect1.y=0; ROI_rect1.width=nCharWidth; ROI_rect1.height=nHeight; cvSetImageROI(pImgResize,ROI_rect1); cvCopy(pImgResize,pImgCharThree,NULL); //获取第3个字符 cvResetImageROI(pImgResize); ROI_rect1.x=col[6]; ROI_rect1.y=0; ROI_rect1.width=nCharWidth; ROI_rect1.height=nHeight; cvSetImageROI(pImgResize,ROI_rect1); cvCopy(pImgResize,pImgCharFour,NULL); //获取第4个字符 cvResetImageROI(pImgResize); ROI_rect1.x=col[8]; ROI_rect1.y=0; ROI_rect1.width=nCharWidth; ROI_rect1.height=nHeight; cvSetImageROI(pImgResize,ROI_rect1); cvCopy(pImgResize,pImgCharFive,NULL); //获取第5个字符 cvResetImageROI(pImgResize); ROI_rect1.x=col[10]; ROI_rect1.y=0; ROI_rect1.width=nCharWidth; ROI_rect1.height=nHeight; cvSetImageROI(pImgResize,ROI_rect1); cvCopy(pImgResize,pImgCharSix,NULL); //获取第6个字符 cvResetImageROI(pImgResize); ROI_rect1.x=col[12]; ROI_rect1.y=0; ROI_rect1.width=nCharWidth; ROI_rect1.height=nHeight; cvSetImageROI(pImgResize,ROI_rect1); cvCopy(pImgResize,pImgCharSeven,NULL); //获取第7个字符 cvResetImageROI(pImgResize); /*cvLine(src,cvPoint(col_start,row_start),cvPoint(col_end,row_start),cvScalar(255,0,0),3,8,0); //在原图中得到车牌区域 cvLine(src,cvPoint(col_start,row_end),cvPoint(col_end,row_end),cvScalar(255,0,0),3,8,0); cvLine(src,cvPoint(col_start,row_start),cvPoint(col_start,row_end),cvScalar(255,0,0),3,8,0); cvLine(src,cvPoint(col_end,row_start),cvPoint(col_end,row_end),cvScalar(255,0,0),3,8,0);*/ pImgHist=cvCreateImage(cvSize(300,200),8,1); //画直方图 CvHistogram *hist=cvCreateHist(1,&hist_size,CV_HIST_ARRAY,ranges,1); cvCalcHist(&pImg8uSmooth,hist,0,0); cvGetMinMaxHistValue(hist,&min_value,&max_value,&min_dx,&max_dx); cout<<min_value<<" "<<max_value<<" "<<min_dx<<" "<<max_dx<<endl; cvScale(hist->bins,hist->bins,((double)pImgHist->height)/max_value,0); cvSet(pImgHist,cvScalarAll(255),0); bin_w=cvRound((pImgHist->width)/hist_size); for(i=0;i<hist_size;i++) { cvRectangle(pImgHist,cvPoint(i*bin_w,pImgHist->height), cvPoint((i+1)*bin_w,pImgHist->height-cvRound(cvGetReal1D(hist->bins,i))), cvScalarAll(0),-1,8,0); } CvMemStorage *storage=NULL; CvSeq *lines=NULL; storage=cvCreateMemStorage(0); lines=cvHoughLines2(pImgCanny,storage,CV_HOUGH_PROBABILISTIC,1,CV_PI/180,80,30,20); /*for(i=0;i<lines->total;i++) { CvPoint *line=(CvPo