From 63ff39f9f3e1e486002266b75410ad988be9e9ec Mon Sep 17 00:00:00 2001 From: Vladislav Vinogradov Date: Wed, 31 Dec 2014 15:37:25 +0300 Subject: [PATCH] remove obsolete gpu optical flow samples --- samples/gpu/brox_optical_flow.cpp | 270 ------------------------------ samples/gpu/performance/tests.cpp | 81 --------- 2 files changed, 351 deletions(-) delete mode 100644 samples/gpu/brox_optical_flow.cpp diff --git a/samples/gpu/brox_optical_flow.cpp b/samples/gpu/brox_optical_flow.cpp deleted file mode 100644 index 638aade45a..0000000000 --- a/samples/gpu/brox_optical_flow.cpp +++ /dev/null @@ -1,270 +0,0 @@ -#include -#include -#include -#include - -#include "opencv2/core.hpp" -#include "opencv2/core/utility.hpp" -#include "opencv2/highgui.hpp" -#include "opencv2/imgproc.hpp" -#include "opencv2/cudaoptflow.hpp" -#include "opencv2/cudaarithm.hpp" - -using namespace std; -using namespace cv; -using namespace cv::cuda; - -void getFlowField(const Mat& u, const Mat& v, Mat& flowField); - -int main(int argc, const char* argv[]) -{ - try - { - const char* keys = - "{ h help | | print help message }" - "{ l left | | specify left image }" - "{ r right | | specify right image }" - "{ s scale | 0.8 | set pyramid scale factor }" - "{ a alpha | 0.197 | set alpha }" - "{ g gamma | 50.0 | set gamma }" - "{ i inner | 10 | set number of inner iterations }" - "{ o outer | 77 | set number of outer iterations }" - "{ si solver | 10 | set number of basic solver iterations }" - "{ t time_step | 0.1 | set frame interpolation time step }"; - - CommandLineParser cmd(argc, argv, keys); - - if (cmd.has("help") || !cmd.check()) - { - cmd.printMessage(); - cmd.printErrors(); - return 0; - } - - string frame0Name = cmd.get("left"); - string frame1Name = cmd.get("right"); - float scale = cmd.get("scale"); - float alpha = cmd.get("alpha"); - float gamma = cmd.get("gamma"); - int inner_iterations = cmd.get("inner"); - int outer_iterations = cmd.get("outer"); - int solver_iterations = cmd.get("solver"); - float timeStep = cmd.get("time_step"); - - if (frame0Name.empty() || frame1Name.empty()) - { - cerr << "Missing input file names" << endl; - return -1; - } - - Mat frame0Color = imread(frame0Name); - Mat frame1Color = imread(frame1Name); - - if (frame0Color.empty() || frame1Color.empty()) - { - cout << "Can't load input images" << endl; - return -1; - } - - cv::cuda::printShortCudaDeviceInfo(cv::cuda::getDevice()); - - cout << "OpenCV / NVIDIA Computer Vision" << endl; - cout << "Optical Flow Demo: Frame Interpolation" << endl; - cout << "=========================================" << endl; - - namedWindow("Forward flow"); - namedWindow("Backward flow"); - - namedWindow("Interpolated frame"); - - cout << "Press:" << endl; - cout << "\tESC to quit" << endl; - cout << "\t'a' to move to the previous frame" << endl; - cout << "\t's' to move to the next frame\n" << endl; - - frame0Color.convertTo(frame0Color, CV_32F, 1.0 / 255.0); - frame1Color.convertTo(frame1Color, CV_32F, 1.0 / 255.0); - - Mat frame0Gray, frame1Gray; - - cv::cvtColor(frame0Color, frame0Gray, COLOR_BGR2GRAY); - cv::cvtColor(frame1Color, frame1Gray, COLOR_BGR2GRAY); - - GpuMat d_frame0(frame0Gray); - GpuMat d_frame1(frame1Gray); - - cout << "Estimating optical flow" << endl; - - BroxOpticalFlow d_flow(alpha, gamma, scale, inner_iterations, outer_iterations, solver_iterations); - - cout << "\tForward..." << endl; - - GpuMat d_fu, d_fv; - - d_flow(d_frame0, d_frame1, d_fu, d_fv); - - Mat flowFieldForward; - getFlowField(Mat(d_fu), Mat(d_fv), flowFieldForward); - - cout << "\tBackward..." << endl; - - GpuMat d_bu, d_bv; - - d_flow(d_frame1, d_frame0, d_bu, d_bv); - - Mat flowFieldBackward; - getFlowField(Mat(d_bu), Mat(d_bv), flowFieldBackward); - - cout << "Interpolating..." << endl; - - // first frame color components - GpuMat d_b, d_g, d_r; - - // second frame color components - GpuMat d_bt, d_gt, d_rt; - - // prepare color components on host and copy them to device memory - Mat channels[3]; - cv::split(frame0Color, channels); - - d_b.upload(channels[0]); - d_g.upload(channels[1]); - d_r.upload(channels[2]); - - cv::split(frame1Color, channels); - - d_bt.upload(channels[0]); - d_gt.upload(channels[1]); - d_rt.upload(channels[2]); - - // temporary buffer - GpuMat d_buf; - - // intermediate frame color components (GPU memory) - GpuMat d_rNew, d_gNew, d_bNew; - - GpuMat d_newFrame; - - vector frames; - frames.reserve(static_cast(1.0f / timeStep) + 2); - - frames.push_back(frame0Color); - - // compute interpolated frames - for (float timePos = timeStep; timePos < 1.0f; timePos += timeStep) - { - // interpolate blue channel - interpolateFrames(d_b, d_bt, d_fu, d_fv, d_bu, d_bv, timePos, d_bNew, d_buf); - - // interpolate green channel - interpolateFrames(d_g, d_gt, d_fu, d_fv, d_bu, d_bv, timePos, d_gNew, d_buf); - - // interpolate red channel - interpolateFrames(d_r, d_rt, d_fu, d_fv, d_bu, d_bv, timePos, d_rNew, d_buf); - - GpuMat channels3[] = {d_bNew, d_gNew, d_rNew}; - cuda::merge(channels3, 3, d_newFrame); - - frames.push_back(Mat(d_newFrame)); - - cout << setprecision(4) << timePos * 100.0f << "%\r"; - } - - frames.push_back(frame1Color); - - cout << setw(5) << "100%" << endl; - - cout << "Done" << endl; - - imshow("Forward flow", flowFieldForward); - imshow("Backward flow", flowFieldBackward); - - int currentFrame = 0; - - imshow("Interpolated frame", frames[currentFrame]); - - for(;;) - { - int key = toupper(waitKey(10) & 0xff); - - switch (key) - { - case 27: - return 0; - - case 'A': - if (currentFrame > 0) - --currentFrame; - - imshow("Interpolated frame", frames[currentFrame]); - break; - - case 'S': - if (currentFrame < static_cast(frames.size()) - 1) - ++currentFrame; - - imshow("Interpolated frame", frames[currentFrame]); - break; - } - } - } - catch (const exception& ex) - { - cerr << ex.what() << endl; - return -1; - } - catch (...) - { - cerr << "Unknow error" << endl; - return -1; - } -} - -template inline T clamp (T x, T a, T b) -{ - return ((x) > (a) ? ((x) < (b) ? (x) : (b)) : (a)); -} - -template inline T mapValue(T x, T a, T b, T c, T d) -{ - x = clamp(x, a, b); - return c + (d - c) * (x - a) / (b - a); -} - -void getFlowField(const Mat& u, const Mat& v, Mat& flowField) -{ - float maxDisplacement = 1.0f; - - for (int i = 0; i < u.rows; ++i) - { - const float* ptr_u = u.ptr(i); - const float* ptr_v = v.ptr(i); - - for (int j = 0; j < u.cols; ++j) - { - float d = max(fabsf(ptr_u[j]), fabsf(ptr_v[j])); - - if (d > maxDisplacement) - maxDisplacement = d; - } - } - - flowField.create(u.size(), CV_8UC4); - - for (int i = 0; i < flowField.rows; ++i) - { - const float* ptr_u = u.ptr(i); - const float* ptr_v = v.ptr(i); - - - Vec4b* row = flowField.ptr(i); - - for (int j = 0; j < flowField.cols; ++j) - { - row[j][0] = 0; - row[j][1] = static_cast (mapValue (-ptr_v[j], -maxDisplacement, maxDisplacement, 0.0f, 255.0f)); - row[j][2] = static_cast (mapValue ( ptr_u[j], -maxDisplacement, maxDisplacement, 0.0f, 255.0f)); - row[j][3] = 255; - } - } -} diff --git a/samples/gpu/performance/tests.cpp b/samples/gpu/performance/tests.cpp index 14910f9a38..9e8bfd28cd 100644 --- a/samples/gpu/performance/tests.cpp +++ b/samples/gpu/performance/tests.cpp @@ -1187,87 +1187,6 @@ TEST(GoodFeaturesToTrack) CUDA_OFF; } -TEST(PyrLKOpticalFlow) -{ - Mat frame0 = imread(abspath("../data/rubberwhale1.png")); - if (frame0.empty()) throw runtime_error("can't open ../data/rubberwhale1.png"); - - Mat frame1 = imread(abspath("../data/rubberwhale2.png")); - if (frame1.empty()) throw runtime_error("can't open ../data/rubberwhale2.png"); - - Mat gray_frame; - cvtColor(frame0, gray_frame, COLOR_BGR2GRAY); - - for (int points = 1000; points <= 8000; points *= 2) - { - SUBTEST << points; - - vector pts; - goodFeaturesToTrack(gray_frame, pts, points, 0.01, 0.0); - - vector nextPts; - vector status; - - vector err; - - calcOpticalFlowPyrLK(frame0, frame1, pts, nextPts, status, err); - - CPU_ON; - calcOpticalFlowPyrLK(frame0, frame1, pts, nextPts, status, err); - CPU_OFF; - - cuda::PyrLKOpticalFlow d_pyrLK; - - cuda::GpuMat d_frame0(frame0); - cuda::GpuMat d_frame1(frame1); - - cuda::GpuMat d_pts; - Mat pts_mat(1, (int)pts.size(), CV_32FC2, (void*)&pts[0]); - d_pts.upload(pts_mat); - - cuda::GpuMat d_nextPts; - cuda::GpuMat d_status; - cuda::GpuMat d_err; - - d_pyrLK.sparse(d_frame0, d_frame1, d_pts, d_nextPts, d_status, &d_err); - - CUDA_ON; - d_pyrLK.sparse(d_frame0, d_frame1, d_pts, d_nextPts, d_status, &d_err); - CUDA_OFF; - } -} - - -TEST(FarnebackOpticalFlow) -{ - const string datasets[] = {"../data/rubberwhale", "../data/basketball"}; - for (size_t i = 0; i < sizeof(datasets)/sizeof(*datasets); ++i) { - for (int fastPyramids = 0; fastPyramids < 2; ++fastPyramids) { - for (int useGaussianBlur = 0; useGaussianBlur < 2; ++useGaussianBlur) { - - SUBTEST << "dataset=" << datasets[i] << ", fastPyramids=" << fastPyramids << ", useGaussianBlur=" << useGaussianBlur; - Mat frame0 = imread(abspath(datasets[i] + "1.png"), IMREAD_GRAYSCALE); - Mat frame1 = imread(abspath(datasets[i] + "2.png"), IMREAD_GRAYSCALE); - if (frame0.empty()) throw runtime_error("can't open " + datasets[i] + "1.png"); - if (frame1.empty()) throw runtime_error("can't open " + datasets[i] + "2.png"); - - cuda::FarnebackOpticalFlow calc; - calc.fastPyramids = fastPyramids != 0; - calc.flags |= useGaussianBlur ? OPTFLOW_FARNEBACK_GAUSSIAN : 0; - - cuda::GpuMat d_frame0(frame0), d_frame1(frame1), d_flowx, d_flowy; - CUDA_ON; - calc(d_frame0, d_frame1, d_flowx, d_flowy); - CUDA_OFF; - - Mat flow; - CPU_ON; - calcOpticalFlowFarneback(frame0, frame1, flow, calc.pyrScale, calc.numLevels, calc.winSize, calc.numIters, calc.polyN, calc.polySigma, calc.flags); - CPU_OFF; - - }}} -} - #ifdef HAVE_OPENCV_BGSEGM TEST(MOG)