Gunnar Farnebak的算法计算稠密光流

最新推荐文章于 2022-11-11 22:03:32 发布

阳光下的Smiles

最新推荐文章于 2022-11-11 22:03:32 发布

阅读量2.5k

点赞数 3

分类专栏：图像处理（opencv）

图像处理（opencv）专栏收录该内容

81 篇文章 9 订阅

订阅专栏

Gunnar Farnebak的算法计算稠密光流

Computes a dense optical flow using the Gunnar Farneback’s algorithm.

用Gunnar Farnebak的算法计算稠密光流

C++: void calcOpticalFlowFarneback (InputArray prev, InputArray next, InputOutputArray flow, double pyr_scale, int levels, int winsize, int iterations, int poly_n, double poly_sigma, int flags ) 示例：

#include "opencv2/video/tracking.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
#include <iostream>
using namespace cv;
using namespace std;


static void help()
{
	cout <<"\nThis program demonstrates dense optical flow algorithm by Gunnar Farneback\n"
	"Mainly the function: calcOpticalFlowFarneback()\n""Call:\n"

	"./fback\n""This reads from video camera 0\n" << 	endl;
}


static void drawOptFlowMap(const Mat& flow, Mat& cflowmap, int step,double, const Scalar& color)
{
	for(int y = 0; y < cflowmap.rows; y += step)
		for(int x = 0; x < cflowmap.cols; x += step)
		{
			const Point2f& fxy = flow.at<Point2f>(y, x);
			line(cflowmap, Point(x,y), Point(cvRound(x+fxy.x), cvRound(y+fxy.y)),color);
			circle(cflowmap, Point(x,y), 2, color, -1);
		}
}

int main(int, char**)
{
	//VideoCapture cap(0);
	VideoCapture cap("E:\\图片\\视频材料\\Megamind.avi");
	help();
	if( !cap.isOpened() )
		return -1;
	Mat prevgray, gray, flow, cflow, frame;
	namedWindow("flow", 1);
	for(;;)
	{
		cap >> frame;
		cvtColor(frame, gray, COLOR_BGR2GRAY);
		if( prevgray.data )
		{
			calcOpticalFlowFarneback(prevgray, gray, flow, 0.5, 3, 15, 3, 5, 1.2, 0);
			cvtColor(prevgray, cflow, COLOR_GRAY2BGR);
			drawOptFlowMap(flow, cflow, 16, 1.5, Scalar(0, 255, 0));
			imshow("flow", cflow);
		}
		if(waitKey(30)>=0)
			break;
		std::swap(prevgray, gray);
	}
	return 0;
}