LK光流法

#include<cv.h>
#include<highgui.h>
#include<iostream>
using namespace std;
using namespace cv;

int main(int argc,char* argv[])
{
	Mat pre,next,frame;
	VideoCapture cap;
	const char *videoFileName = "optical_flow_input.avi";
	//cap.open(0);
	cap.open(videoFileName);

	vector<Point2f> prepoint,nextpoint;
	vector<uchar> state;
	vector<float>err;
	if(!cap.isOpened())
	{
		//cout<<"摄像头未打开，请打开摄像头！！！"<<endl;
		cout<<"Open video file "<<videoFileName<<" failed!\n";
	}

	for(;;)
	{
		cap>>frame;
		if(frame.empty())break;

		cvtColor(frame,next,CV_BGR2GRAY);

		if(!next.empty()&&!pre.empty())
		{

			goodFeaturesToTrack(pre,prepoint,500,0.001,10,Mat(),3,false,0.04);
			cornerSubPix(pre,prepoint,Size(10,10),Size(-1,-1),TermCriteria(CV_TERMCRIT_ITER|CV_TERMCRIT_EPS, 20, 0.03));
			calcOpticalFlowPyrLK(pre,next,prepoint,nextpoint,state,err,Size(31,31),3);
			for(int i=0;i<state.size();i++)
			{
				if(state[i]!=0)
				{
					line(frame,Point((int)prepoint[i].x,(int)prepoint[i].y),Point((int)nextpoint[i].x,(int)nextpoint[i].y),Scalar(0,255,0));
				}
			}
			namedWindow("frame",0);
			imshow("frame",frame);
			waitKey(1);
		}
		next.copyTo(pre);
	}
}