1 #include "opencv2/video/tracking_c.h"
2 #include "opencv2/highgui/highgui_c.h"
3 #include "opencv2/imgproc/imgproc_c.h"
9 "\n This program demonstrate dense \"Farneback\n optical flow\n"
10 "It read from camera 0, and shows how to use and display dense Franeback optical flow\n"
16 static void drawOptFlowMap(const CvMat* flow, CvMat* cflowmap, int step,
17 double scale, CvScalar color)
21 for( y = 0; y < cflowmap->rows; y += step)
22 for( x = 0; x < cflowmap->cols; x += step)
24 CvPoint2D32f fxy = CV_MAT_ELEM(*flow, CvPoint2D32f, y, x);
25 cvLine(cflowmap, cvPoint(x,y), cvPoint(cvRound(x+fxy.x), cvRound(y+fxy.y)),
27 cvCircle(cflowmap, cvPoint(x,y), 2, color, -1, 8, 0);
31 int main( int argc, char** argv )
33 CvCapture* capture = cvCreateCameraCapture(0);
34 CvMat* prevgray = 0, *gray = 0, *flow = 0, *cflow = 0;
35 (void)argc; (void)argv;
42 cvNamedWindow("flow", 1);
46 int firstFrame = gray == 0;
47 IplImage* frame = cvQueryFrame(capture);
52 gray = cvCreateMat(frame->height, frame->width, CV_8UC1);
53 prevgray = cvCreateMat(gray->rows, gray->cols, gray->type);
54 flow = cvCreateMat(gray->rows, gray->cols, CV_32FC2);
55 cflow = cvCreateMat(gray->rows, gray->cols, CV_8UC3);
57 cvCvtColor(frame, gray, CV_BGR2GRAY);
61 cvCalcOpticalFlowFarneback(prevgray, gray, flow, 0.5, 3, 15, 3, 5, 1.2, 0);
62 cvCvtColor(prevgray, cflow, CV_GRAY2BGR);
63 drawOptFlowMap(flow, cflow, 16, 1.5, CV_RGB(0, 255, 0));
64 cvShowImage("flow", cflow);
70 CV_SWAP(prevgray, gray, temp);
73 cvReleaseCapture(&capture);