三种光流算法的实现源码及测试结果.docx

基于OpenCV的三种光流算法实现源码及测试结果本文包括三种基于OpenCV的光流算法实现源码及测试结果。具体为HS算法，LK算法，和ctfLK算法，算法的原实现作者是Eric Yuan，这里是作者的博客主页：http://eric-yuan.me。本文对这三种光流算法进行了相关调试及结果验证，供大家在自己的项目开发中参考。第一种：HS光流法（作者HORN 和SCHUNCK）#includeopencv2/core/core.hpp#includeopencv2/imgproc/imgproc.hpp#includeopencv2/highgui/highgui.hpp#includemath.h#includefstream#includeiostreamusingnamespace cv;usingnamespace std;#define ATD atdouble#define elif elseif#ifndefbool#defineboolint#definefalse ((bool)0)#definetrue ((bool)1)#endifMat get_fx(Mat src1, Mat src2){Mat fx;Mat kernel = Mat::ones(2, 2, CV_64FC1);kernel.ATD(0, 0) = -1.0;kernel.ATD(1, 0) = -1.0;Mat dst1, dst2;filter2D(src1, dst1, -1, kernel);filter2D(src2, dst2, -1, kernel);fx = dst1 + dst2;return fx;}Mat get_fy(Mat src1, Mat src2){Mat fy;Mat kernel = Mat::ones(2, 2, CV_64FC1);kernel.ATD(0, 0) = -1.0;kernel.ATD(0, 1) = -1.0;Mat dst1, dst2;filter2D(src1, dst1, -1, kernel);filter2D(src2, dst2, -1, kernel);fy = dst1 + dst2;return fy;}Mat get_ft(Mat src1, Mat src2){Mat ft;Mat kernel = Mat::ones(2, 2, CV_64FC1);kernel = kernel.mul(-1);Mat dst1, dst2;filter2D(src1, dst1, -1, kernel);kernel = kernel.mul(-1);filter2D(src2, dst2, -1, kernel);ft = dst1 + dst2;return ft;}bool isInsideImage(int y, int x, Mat m){int width = m.cols;int height = m.rows;if (x = 0 x width y = 0 y height) returntrue;elsereturnfalse;}double get_Average4(Mat m, int y, int x){if (x 0 || x = m.cols) return 0;if (y 0 || y = m.rows) return 0;double val = 0.0;int tmp = 0;if (isInsideImage(y - 1, x, m)){++tmp;val += m.ATD(y - 1, x);}if (isInsideImage(y + 1, x, m)){++tmp;val += m.ATD(y + 1, x);}if (isInsideImage(y, x - 1, m)){++tmp;val += m.ATD(y, x - 1);}if (isInsideImage(y, x + 1, m)){++tmp;val += m.ATD(y, x + 1);}return val / tmp;}Mat get_Average4_Mat(Mat m){Mat res = Mat::zeros(m.rows, m.cols, CV_64FC1);for (int i = 0; i m.rows; i++){for (int j = 0; j m.cols; j++){res.ATD(i, j) = get_Average4(m, i, j);}}return res;}void saveMat(Mat M, string s){s += .txt;FILE *pOut = fopen(s.c_str(), w+);f