opencv图像处理5--车辆检测

ゝ一世哀愁。 2022-06-18 04:39 153阅读 0赞

void CTrafficsampleDlg::OnOK() //读入视频并检测车辆  
\{

// TODO: Add extra validation here

> //此处添加图像处理的代码

//CDialog::OnOK();  
IplImage \*frame = NULL;  //定义帧,每帧就是一张图    
IplImage \*B\_part = NULL, \*G\_part = NULL, \*R\_part = NULL;  
IplImage \*pFrImg = NULL; //前景图片  
IplImage \*pBkImg = NULL; //背景图片  
IplImage \*dst = NULL;  
  
  
CvMemStorage\* storage = cvCreateMemStorage(0);  
CvSeq\* contour = 0;  
   
//轮廓矩形  
CvRect contourRect = cvRect(0,0,0,0);  
  
  
//可以获取视频的各种属性  
CvCapture \* pCapture = cvCreateFileCapture("traffic.avi");  
  
  
cvNamedWindow("Camera",CV\_WINDOW\_AUTOSIZE);  //设置窗口    
cvNamedWindow("temp", CV\_WINDOW\_AUTOSIZE);   
cvNamedWindow("background",1);   
cvNamedWindow("foreground",1);  
cvNamedWindow("contours",1);  
     
cvMoveWindow("background", 360, 0);   
cvMoveWindow("foreground", 690, 0);  
  
  
//初始化高斯混合模型参数  
CvGaussBGModel \*bg\_model=NULL;  
  
  
  
int count = 1;  
while (frame = cvQueryFrame(pCapture))        
\{      
if(count ==1)  
\{  
B\_part = cvCreateImage(cvGetSize(frame),frame->depth,1);  
G\_part = cvCreateImage(cvGetSize(frame),frame->depth,1);  
R\_part = cvCreateImage(cvGetSize(frame),frame->depth,1);  
pBkImg = cvCreateImage(cvSize(frame->width, frame->height),  IPL\_DEPTH\_8U,3);     
pFrImg = cvCreateImage(cvSize(frame->width, frame->height),  IPL\_DEPTH\_8U,1);     
dst = cvCreateImage( cvGetSize(frame), 8, 3 );//找到轮廓并绘制轮廓  
bg\_model = (CvGaussBGModel\*)cvCreateGaussianBGModel(frame, 0);  
\}  
else  
\{  
cvUpdateBGStatModel(frame, (CvBGStatModel \*)bg\_model );//更新高斯背景模型  
  
//pFrImg为前景图像，只能为单通道  
//pBkImg为背景图像，可以为单通道或与pFrame通道数相同  
cvCopy(bg\_model->foreground,pFrImg,0);  
cvCopy(bg\_model->background,pBkImg,0);  
//把图像正过来  
pBkImg->origin=1;  
pFrImg->origin=1;  
cvSplit(frame , B\_part,G\_part, R\_part,0);  
cvFlip(B\_part,B\_part,0);//沿X轴翻转一次  
  
  
//下面进行阈值分割  
//如何确定阈值？？是否存在编好的函数呢？  
//固定阈值分割方式，会导致一张图片中如果亮暗不均，一张图的不同区域分割效果差距很大  
//如果能够提前提取出图像的背景，然后想减的方式会使分割效果变好  
cvThreshold(B\_part,B\_part, 195, 255,CV\_THRESH\_BINARY);  
  
  
//阈值分割之后，进行膨胀操作和腐蚀操作  
//默认的模板由于腐蚀和膨胀的幅度比较大，效果不是很好啊  
//最好是自定义模板  
//观察白线大多为竖直的，腐蚀时主要是横向腐蚀  
//车体车头和车身由于在前玻璃的分割作用，数要是在竖向，膨胀时主要在竖向，但是需避免造成两车的太近而融为一体  
IplConvKernel \* kenel = cvCreateStructuringElementEx(3,1,1,0,CV\_SHAPE\_RECT,NULL);  
IplConvKernel \* kenel2 = cvCreateStructuringElementEx(1,3,0,1,CV\_SHAPE\_RECT,NULL);  
  
cvErode(B\_part, B\_part,kenel,1 );  
cvDilate(B\_part,B\_part,kenel,1);  
cvErode(B\_part, B\_part,kenel2, 1);  
vErode(B\_part, B\_part,kenel,1 );  
    
cvDilate(B\_part,B\_part,kenel2,2);  
cvDilate(B\_part,B\_part,kenel2,2);  
cvDilate(B\_part,B\_part,NULL,1);  
  
  
//获取并绘制轮廓    
cvFindContours( B\_part, storage, &contour, sizeof(CvContour), CV\_RETR\_CCOMP, CV\_CHAIN\_APPROX\_SIMPLE );  
cvZero( dst );  
for( ; contour != 0; contour = contour->h\_next )  
\{  
  
  
contourRect = cvBoundingRect(contour,0);  
  
  
if((contourRect.width > 10) &&(contourRect.height >12))\{  
CvScalar color = CV\_RGB( rand()&255, rand()&255, rand()&255 );//每次都随机颜色显示  
/\* 用1替代 CV\_FILLED  所指示的轮廓外形 \*/  
cvDrawContours( dst, contour, color, color, -1, CV\_FILLED, 8 );  
//cvRectangle(pFrame,pt3,pt4,CV\_RGB(255,0,0),1, 8, 0 );  
cvRectangle(dst ,cvPoint(contourRect.x, contourRect.y),  
cvPoint(contourRect.x + contourRect.width, contourRect.y + contourRect.height),  
CV\_RGB(255,0,0),  
1,  
8,  
0);  
\}  
   
\}  
   
  ![0_1316929681lH7m.gif][] ![0_1316929794cws2.gif][]  
cvShowImage("Camera", frame);    
cvShowImage("temp" ,B\_part);  
cvShowImage("background", pBkImg);      
cvShowImage("foreground", pFrImg);    
cvShowImage( "contours", dst );  
  
  
int key1;  
key1 = cvWaitKey (300);  
if (key1 == 'q' || key1 == 'Q')    
break;    
  
  
\}  
count++;  
\}     
  
  
  
    cvDestroyWindow("temp");  
cvReleaseCapture(&pCapture); //销毁摄像头    
cvDestroyWindow("Camera");  //销毁窗口   
cvReleaseBGStatModel((CvBGStatModel\*\*)&bg\_model);  
cvDestroyWindow("background");   
cvDestroyWindow("foreground");     
cvReleaseImage(&pFrImg);   
cvReleaseImage(&pBkImg);     
cvReleaseImage(&dst);  
cvDestroyWindow("contours");  
//CTrafficsampleDlg::OnOK();  
\}

[0_1316929681lH7m.gif]: http://hi.csdn.net/attachment/201109/25/0_1316929681lH7m.gif
[0_1316929794cws2.gif]: /images/20220618/32a64b2bf5c64293b2a8ebd7cbc576a5.png