motiondetect.c
上传用户:banwdc
上传日期:2016-06-25
资源大小:2871k
文件大小:5k
源码类别:

OpenCV

开发平台:

Visual C++

  1. #include "cv.h"
  2. #include "highgui.h"
  3. #include <time.h>
  4. #include <math.h>
  5. #include <ctype.h>
  6. #include <stdio.h>
  7. #include <string.h>
  8. // various tracking parameters (in seconds)
  9. const double MHI_DURATION = 0.5;
  10. const double MAX_TIME_DELTA = 0.5;
  11. const double MIN_TIME_DELTA = 0.05;
  12. const int N = 3;
  13. //
  14. const int CONTOUR_MAX_AERA = 16;
  15. // ring image buffer
  16. IplImage **buf = 0;
  17. int last = 0;
  18. // temporary images
  19. IplImage *mhi = 0; // MHI: motion history image
  20. CvFilter filter = CV_GAUSSIAN_5x5;
  21. CvConnectedComp *cur_comp, min_comp;
  22. CvConnectedComp comp;
  23. CvMemStorage *storage;
  24. CvPoint pt[4];
  25. //  参数:
  26. //  img – 输入视频帧
  27. //  dst – 检测结果
  28. void  update_mhi( IplImage* img, IplImage* dst, int diff_threshold )
  29. {
  30.     double timestamp = clock()/100.; // get current time in seconds
  31.     CvSize size = cvSize(img->width,img->height); // get current frame size
  32.     int i, j, idx1, idx2;
  33.     IplImage* silh;
  34.     uchar val;
  35.     float temp;
  36.     IplImage* pyr = cvCreateImage( cvSize((size.width & -2)/2, (size.height & -2)/2), 8, 1 );
  37.     CvMemStorage *stor;
  38.     CvSeq *cont, *result, *squares;
  39.     CvSeqReader reader;
  40.     if( !mhi || mhi->width != size.width || mhi->height != size.height ) 
  41.     {
  42.         if( buf == 0 ) 
  43.         {
  44.             buf = (IplImage**)malloc(N*sizeof(buf[0]));
  45.             memset( buf, 0, N*sizeof(buf[0]));
  46.         }
  47.         
  48.         for( i = 0; i < N; i++ ) 
  49.         {
  50.             cvReleaseImage( &buf[i] );
  51.             buf[i] = cvCreateImage( size, IPL_DEPTH_8U, 1 );
  52.             cvZero( buf[i] );
  53.         }
  54.         cvReleaseImage( &mhi );
  55.         mhi = cvCreateImage( size, IPL_DEPTH_32F, 1 );
  56.         cvZero( mhi ); // clear MHI at the beginning
  57.     } // end of if(mhi)
  58.     cvCvtColor( img, buf[last], CV_BGR2GRAY ); // convert frame to grayscale
  59.     idx1 = last;
  60.     idx2 = (last + 1) % N; // index of (last - (N-1))th frame 
  61.     last = idx2;
  62.     // 做帧差
  63.     silh = buf[idx2];
  64.     cvAbsDiff( buf[idx1], buf[idx2], silh ); // get difference between frames
  65.     
  66.     // 对差图像做二值化
  67.     cvThreshold( silh, silh, 30, 255, CV_THRESH_BINARY ); // and threshold it
  68.     
  69.     cvUpdateMotionHistory( silh, mhi, timestamp, MHI_DURATION ); // update MHI
  70.     cvCvtScale( mhi, dst, 255./MHI_DURATION, 
  71.       (MHI_DURATION - timestamp)*255./MHI_DURATION );    
  72.     cvCvtScale( mhi, dst, 255./MHI_DURATION, 0 );    
  73.     
  74.     // 中值滤波,消除小的噪声
  75.     cvSmooth( dst, dst, CV_MEDIAN, 3, 0, 0, 0 );
  76.     
  77.     // 向下采样,去掉噪声
  78.     cvPyrDown( dst, pyr, 7 );
  79.     cvDilate( pyr, pyr, 0, 1 );  // 做膨胀操作,消除目标的不连续空洞
  80.     cvPyrUp( pyr, dst, 7 );
  81.     //
  82.     // 下面的程序段用来找到轮廓
  83.     //
  84.     // Create dynamic structure and sequence.
  85.     stor = cvCreateMemStorage(0);
  86.     cont = cvCreateSeq(CV_SEQ_ELTYPE_POINT, sizeof(CvSeq), sizeof(CvPoint) , stor);
  87.     
  88.     // 找到所有轮廓
  89.     cvFindContours( dst, stor, &cont, sizeof(CvContour), 
  90.                     CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE, cvPoint(0,0));
  91. /*
  92.     for(;cont;cont = cont->h_next)
  93.     {   
  94.         // Number point must be more than or equal to 6 (for cvFitEllipse_32f).        
  95.         if( cont->total < 6 )  
  96.             continue;
  97.         // Draw current contour.
  98.         cvDrawContours(img,cont,CV_RGB(255,0,0),CV_RGB(255,0,0),0,1, 8, cvPoint(0,0));
  99.     }  // end of for-loop: "cont"
  100. */
  101.     // 直接使用CONTOUR中的矩形来画轮廓
  102.     for(;cont;cont = cont->h_next)
  103.     {
  104.               CvRect r = ((CvContour*)cont)->rect;
  105.               if(r.height * r.width > CONTOUR_MAX_AERA) // 面积小的方形抛弃掉
  106.               {
  107.                   cvRectangle( img, cvPoint(r.x,r.y), 
  108.                           cvPoint(r.x + r.width, r.y + r.height),
  109.                           CV_RGB(255,0,0), 1, CV_AA,0);
  110.               }
  111.     }
  112.     // free memory
  113.     cvReleaseMemStorage(&stor);
  114.     cvReleaseImage( &pyr );
  115. }
  116. int main(int argc, char** argv)
  117. {
  118.     IplImage* motion = 0; 
  119.     CvCapture* capture = 0;
  120.     
  121.     if( argc == 1 || (argc == 2 && strlen(argv[1]) == 1 && isdigit(argv[1][0])))
  122.         capture = cvCaptureFromCAM( argc == 2 ? argv[1][0] - '0' : 0 );
  123.     else if( argc == 2 )
  124.         capture = cvCaptureFromAVI( argv[1] ); 
  125.     if( capture )
  126.     {
  127.         cvNamedWindow( "Motion", 1 );
  128.         for(;;)
  129.         {
  130.             IplImage* image;
  131.             if( !cvGrabFrame( capture ))
  132.                 break;
  133.             image = cvRetrieveFrame( capture );
  134.             if( image )
  135.             {
  136.                 if( !motion )
  137.                 {
  138.                     motion = cvCreateImage( cvSize(image->width,image->height), 8, 1 );
  139.                     cvZero( motion );
  140.                     motion->origin = image->origin;
  141.                 }
  142.             }
  143.             update_mhi( image, motion, 60 );
  144.             cvShowImage( "Motion", image );
  145.             if( cvWaitKey(10) >= 0 )
  146.                 break;
  147.         }
  148.         cvReleaseCapture( &capture );
  149.         cvDestroyWindow( "Motion" );
  150.     }
  151.     return 0;
  152. }