mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.3 KiB
66 lines
2.3 KiB
// This file is part of OpenCV project. |
|
// It is subject to the license terms in the LICENSE file found in the top-level directory |
|
// of this distribution and at http://opencv.org/license.html. |
|
#include "perf_precomp.hpp" |
|
|
|
namespace opencv_test { namespace { |
|
|
|
void MakeArtificialExample(Mat &dst_frame1, Mat &dst_frame2); |
|
|
|
typedef tuple<String, Size> DISParams; |
|
typedef TestBaseWithParam<DISParams> DenseOpticalFlow_DIS; |
|
|
|
PERF_TEST_P(DenseOpticalFlow_DIS, perf, |
|
Combine(Values("PRESET_ULTRAFAST", "PRESET_FAST", "PRESET_MEDIUM"), Values(szVGA, sz720p, sz1080p))) |
|
{ |
|
DISParams params = GetParam(); |
|
|
|
// use strings to print preset names in the perf test results: |
|
String preset_string = get<0>(params); |
|
int preset = DISOpticalFlow::PRESET_FAST; |
|
if (preset_string == "PRESET_ULTRAFAST") |
|
preset = DISOpticalFlow::PRESET_ULTRAFAST; |
|
else if (preset_string == "PRESET_FAST") |
|
preset = DISOpticalFlow::PRESET_FAST; |
|
else if (preset_string == "PRESET_MEDIUM") |
|
preset = DISOpticalFlow::PRESET_MEDIUM; |
|
Size sz = get<1>(params); |
|
|
|
Mat frame1(sz, CV_8U); |
|
Mat frame2(sz, CV_8U); |
|
Mat flow; |
|
|
|
MakeArtificialExample(frame1, frame2); |
|
|
|
TEST_CYCLE_N(10) |
|
{ |
|
Ptr<DenseOpticalFlow> algo = DISOpticalFlow::create(preset); |
|
algo->calc(frame1, frame2, flow); |
|
} |
|
|
|
SANITY_CHECK_NOTHING(); |
|
} |
|
|
|
void MakeArtificialExample(Mat &dst_frame1, Mat &dst_frame2) |
|
{ |
|
int src_scale = 2; |
|
int OF_scale = 6; |
|
double sigma = dst_frame1.cols / 300; |
|
|
|
Mat tmp(Size(dst_frame1.cols / (1 << src_scale), dst_frame1.rows / (1 << src_scale)), CV_8U); |
|
randu(tmp, 0, 255); |
|
resize(tmp, dst_frame1, dst_frame1.size(), 0.0, 0.0, INTER_LINEAR_EXACT); |
|
resize(tmp, dst_frame2, dst_frame2.size(), 0.0, 0.0, INTER_LINEAR_EXACT); |
|
|
|
Mat displacement_field(Size(dst_frame1.cols / (1 << OF_scale), dst_frame1.rows / (1 << OF_scale)), |
|
CV_32FC2); |
|
randn(displacement_field, 0.0, sigma); |
|
resize(displacement_field, displacement_field, dst_frame2.size(), 0.0, 0.0, INTER_CUBIC); |
|
for (int i = 0; i < displacement_field.rows; i++) |
|
for (int j = 0; j < displacement_field.cols; j++) |
|
displacement_field.at<Vec2f>(i, j) += Vec2f((float)j, (float)i); |
|
|
|
remap(dst_frame2, dst_frame2, displacement_field, Mat(), INTER_LINEAR, BORDER_REPLICATE); |
|
} |
|
|
|
}} // namespace
|
|
|