"SimpleFlow" optical flow estimation algorithm (GSoC project) declaration in includes, implementation, usage example, test
This commit is contained in:
Yury Zemlyanskiy
committed by
Vadim Pisarevsky
Vadim Pisarevsky
parent
2367a195c5
commit
c77d099018
193
modules/video/test/test_simpleflow.cpp
Normal file
193
modules/video/test/test_simpleflow.cpp
Normal file
@@ -0,0 +1,193 @@
|
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// Intel License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2000, Intel Corporation, all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
//
|
||||
// * The name of Intel Corporation may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#include "test_precomp.hpp"
|
||||
|
||||
#include <string>
|
||||
|
||||
using namespace std;
|
||||
|
||||
/* ///////////////////// simpleflow_test ///////////////////////// */
|
||||
|
||||
class CV_SimpleFlowTest : public cvtest::BaseTest
|
||||
{
|
||||
public:
|
||||
CV_SimpleFlowTest();
|
||||
protected:
|
||||
void run(int);
|
||||
};
|
||||
|
||||
|
||||
CV_SimpleFlowTest::CV_SimpleFlowTest() {}
|
||||
|
||||
static void readOpticalFlowFromFile(FILE* file, cv::Mat& flowX, cv::Mat& flowY) {
|
||||
char header[5];
|
||||
if (fread(header, 1, 4, file) < 4 && (string)header != "PIEH") {
|
||||
return;
|
||||
}
|
||||
|
||||
int cols, rows;
|
||||
if (fread(&cols, sizeof(int), 1, file) != 1||
|
||||
fread(&rows, sizeof(int), 1, file) != 1) {
|
||||
return;
|
||||
}
|
||||
|
||||
flowX = cv::Mat::zeros(rows, cols, CV_64F);
|
||||
flowY = cv::Mat::zeros(rows, cols, CV_64F);
|
||||
|
||||
for (int i = 0; i < rows; ++i) {
|
||||
for (int j = 0; j < cols; ++j) {
|
||||
float uPoint, vPoint;
|
||||
if (fread(&uPoint, sizeof(float), 1, file) != 1 ||
|
||||
fread(&vPoint, sizeof(float), 1, file) != 1) {
|
||||
flowX.release();
|
||||
flowY.release();
|
||||
return;
|
||||
}
|
||||
|
||||
flowX.at<double>(i, j) = uPoint;
|
||||
flowY.at<double>(i, j) = vPoint;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static bool isFlowCorrect(double u) {
|
||||
return !isnan(u) && (fabs(u) < 1e9);
|
||||
}
|
||||
|
||||
static double calc_rmse(cv::Mat flow1X, cv::Mat flow1Y, cv::Mat flow2X, cv::Mat flow2Y) {
|
||||
long double sum;
|
||||
int counter = 0;
|
||||
const int rows = flow1X.rows;
|
||||
const int cols = flow1X.cols;
|
||||
|
||||
for (int y = 0; y < rows; ++y) {
|
||||
for (int x = 0; x < cols; ++x) {
|
||||
double u1 = flow1X.at<double>(y, x);
|
||||
double v1 = flow1Y.at<double>(y, x);
|
||||
double u2 = flow2X.at<double>(y, x);
|
||||
double v2 = flow2Y.at<double>(y, x);
|
||||
if (isFlowCorrect(u1) && isFlowCorrect(u2) && isFlowCorrect(v1) && isFlowCorrect(v2)) {
|
||||
sum += (u1-u2)*(u1-u2) + (v1-v2)*(v1-v2);
|
||||
counter++;
|
||||
}
|
||||
}
|
||||
}
|
||||
return sqrt((double)sum / (1e-9 + counter));
|
||||
}
|
||||
|
||||
void CV_SimpleFlowTest::run(int) {
|
||||
int code = cvtest::TS::OK;
|
||||
|
||||
const double MAX_RMSE = 0.6;
|
||||
const string frame1_path = ts->get_data_path() + "optflow/RubberWhale1.png";
|
||||
const string frame2_path = ts->get_data_path() + "optflow/RubberWhale2.png";
|
||||
const string gt_flow_path = ts->get_data_path() + "optflow/RubberWhale.flo";
|
||||
|
||||
cv::Mat frame1 = cv::imread(frame1_path);
|
||||
cv::Mat frame2 = cv::imread(frame2_path);
|
||||
|
||||
if (frame1.empty()) {
|
||||
ts->printf(cvtest::TS::LOG, "could not read image %s\n", frame2_path.c_str());
|
||||
ts->set_failed_test_info(cvtest::TS::FAIL_MISSING_TEST_DATA);
|
||||
return;
|
||||
}
|
||||
|
||||
if (frame2.empty()) {
|
||||
ts->printf(cvtest::TS::LOG, "could not read image %s\n", frame2_path.c_str());
|
||||
ts->set_failed_test_info(cvtest::TS::FAIL_MISSING_TEST_DATA);
|
||||
return;
|
||||
}
|
||||
|
||||
if (frame1.rows != frame2.rows && frame1.cols != frame2.cols) {
|
||||
ts->printf(cvtest::TS::LOG, "images should be of equal sizes (%s and %s)",
|
||||
frame1_path.c_str(), frame2_path.c_str());
|
||||
ts->set_failed_test_info(cvtest::TS::FAIL_MISSING_TEST_DATA);
|
||||
return;
|
||||
}
|
||||
|
||||
if (frame1.type() != 16 || frame2.type() != 16) {
|
||||
ts->printf(cvtest::TS::LOG, "images should be of equal type CV_8UC3 (%s and %s)",
|
||||
frame1_path.c_str(), frame2_path.c_str());
|
||||
ts->set_failed_test_info(cvtest::TS::FAIL_MISSING_TEST_DATA);
|
||||
return;
|
||||
}
|
||||
|
||||
cv::Mat flowX_gt, flowY_gt;
|
||||
|
||||
FILE* gt_flow_file = fopen(gt_flow_path.c_str(), "rb");
|
||||
if (gt_flow_file == NULL) {
|
||||
ts->printf(cvtest::TS::LOG, "could not read ground-thuth flow from file %s",
|
||||
gt_flow_path.c_str());
|
||||
ts->set_failed_test_info(cvtest::TS::FAIL_MISSING_TEST_DATA);
|
||||
return;
|
||||
}
|
||||
readOpticalFlowFromFile(gt_flow_file, flowX_gt, flowY_gt);
|
||||
if (flowX_gt.empty() || flowY_gt.empty()) {
|
||||
ts->printf(cvtest::TS::LOG, "error while reading flow data from file %s",
|
||||
gt_flow_path.c_str());
|
||||
ts->set_failed_test_info(cvtest::TS::FAIL_MISSING_TEST_DATA);
|
||||
return;
|
||||
}
|
||||
fclose(gt_flow_file);
|
||||
|
||||
cv::Mat flowX, flowY;
|
||||
cv::calcOpticalFlowSF(frame1, frame2,
|
||||
flowX, flowY,
|
||||
3, 4, 2, 4.1, 25.5, 18, 55.0, 25.5, 0.35, 18, 55.0, 25.5, 10);
|
||||
|
||||
double rmse = calc_rmse(flowX_gt, flowY_gt, flowX, flowY);
|
||||
|
||||
ts->printf(cvtest::TS::LOG, "Optical flow estimation RMSE for SimpleFlow algorithm : %lf\n",
|
||||
rmse);
|
||||
|
||||
if (rmse > MAX_RMSE) {
|
||||
ts->printf( cvtest::TS::LOG,
|
||||
"Too big rmse error : %lf ( >= %lf )\n", rmse, MAX_RMSE);
|
||||
ts->set_failed_test_info(cvtest::TS::FAIL_BAD_ACCURACY);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
TEST(Video_OpticalFlowSimpleFlow, accuracy) { CV_SimpleFlowTest test; test.safe_run(); }
|
||||
|
||||
/* End of file. */
|
||||
Reference in New Issue
Block a user