#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);
}
}
LK光流法
最新推荐文章于 2024-07-05 19:39:03 发布