光流法 \OpenCV2.1\samples\c\lkdemo.exe

来源:互联网 发布:the eye infinite 知乎 编辑:程序博客网 时间:2024/04/28 06:12
/////////////光流法//////////////#include <D:\\OpenCV\\OpenCV2.2\\include\\opencv2/video/tracking.hpp>#include <D:\\OpenCV\\OpenCV2.2\\include\\opencv2/highgui/highgui.hpp>#include <stdio.h>void help(){printf("This program demonstrate dense \"Farneback\n optical flow\n""It read from camera 0, and shows how to use and display dense Franeback optical flow\n""Call:\n""./fback_c\n\n");}void drawOptFlowMap(const CvMat* flow, CvMat* cflowmap, int step,double scale, CvScalar color){int x, y;for( y = 0; y < cflowmap->rows; y += step)for( x = 0; x < cflowmap->cols; x += step){CvPoint2D32f fxy = CV_MAT_ELEM(*flow, CvPoint2D32f, y, x);cvLine(cflowmap, cvPoint(x,y), cvPoint(cvRound(x+fxy.x), cvRound(y+fxy.y)),color, 1, 8, 0);cvCircle(cflowmap, cvPoint(x,y), 2, color, -1, 8, 0);}}int main(int argc, char** argv){CvCapture* capture = cvCreateCameraCapture(0);CvMat* prevgray = 0, *gray = 0, *flow = 0, *cflow = 0;help();if( !capture )return -1;cvNamedWindow("flow", 1);for(;;){int firstFrame = gray == 0;IplImage* frame = cvQueryFrame(capture);if(!frame)break;if(!gray){gray = cvCreateMat(frame->height, frame->width, CV_8UC1);prevgray = cvCreateMat(gray->rows, gray->cols, gray->type);flow = cvCreateMat(gray->rows, gray->cols, CV_32FC2);cflow = cvCreateMat(gray->rows, gray->cols, CV_8UC3);}cvCvtColor(frame, gray, CV_BGR2GRAY);if( !firstFrame ){cvCalcOpticalFlowFarneback(prevgray, gray, flow, 0.5, 3, 15, 3, 5, 1.2, 0);cvCvtColor(prevgray, cflow, CV_GRAY2BGR);drawOptFlowMap(flow, cflow, 16, 1.5, CV_RGB(0, 255, 0));cvShowImage("flow", cflow);}if(cvWaitKey(30)>=0)break;{CvMat* temp;    CV_SWAP(prevgray, gray, temp);}}cvReleaseCapture(&capture);return 0;}

原创粉丝点击