← Back to branch summary

~paparazzi-uav/paparazzi/v5.0-manual

« back to all changes in this revision

Viewing changes to sw/ext/opencv_bebop/opencv/modules/video/test/ocl/test_optflow_farneback.cpp

  • Committer: Paparazzi buildbot
  • Date: 2016-05-18 15:00:29 UTC
  • Revision ID: felix.ruess+docbot@gmail.com-20160518150029-e8lgzi5kvb4p7un9
Manual import commit 4b8bbb730080dac23cf816b98908dacfabe2a8ec from v5.0 branch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
sw/ext/opencv_bebop/opencv/modules/video/test/ocl/test_optflow_farneback.cpp
 
1
/*M///////////////////////////////////////////////////////////////////////////////////////
 
2
//
 
3
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
 
4
//
 
5
//  By downloading, copying, installing or using the software you agree to this license.
 
6
//  If you do not agree to this license, do not download, install,
 
7
//  copy or use the software.
 
8
//
 
9
//
 
10
//                           License Agreement
 
11
//                For Open Source Computer Vision Library
 
12
//
 
13
// Copyright (C) 2010-2012, Institute Of Software Chinese Academy Of Science, all rights reserved.
 
14
// Copyright (C) 2010-2012, Advanced Micro Devices, Inc., all rights reserved.
 
15
// Copyright (C) 2010-2012, Multicoreware, Inc., all rights reserved.
 
16
// Third party copyrights are property of their respective owners.
 
17
//
 
18
// Redistribution and use in source and binary forms, with or without modification,
 
19
// are permitted provided that the following conditions are met:
 
20
//
 
21
//   * Redistribution's of source code must retain the above copyright notice,
 
22
//     this list of conditions and the following disclaimer.
 
23
//
 
24
//   * Redistribution's in binary form must reproduce the above copyright notice,
 
25
//     this list of conditions and the following disclaimer in the documentation
 
26
//     and/or other materials provided with the distribution.
 
27
//
 
28
//   * The name of the copyright holders may not be used to endorse or promote products
 
29
//     derived from this software without specific prior written permission.
 
30
//
 
31
// This software is provided by the copyright holders and contributors "as is" and
 
32
// any express or implied warranties, including, but not limited to, the implied
 
33
// warranties of merchantability and fitness for a particular purpose are disclaimed.
 
34
// In no event shall the Intel Corporation or contributors be liable for any direct,
 
35
// indirect, incidental, special, exemplary, or consequential damages
 
36
// (including, but not limited to, procurement of substitute goods or services;
 
37
// loss of use, data, or profits; or business interruption) however caused
 
38
// and on any theory of liability, whether in contract, strict liability,
 
39
// or tort (including negligence or otherwise) arising in any way out of
 
40
// the use of this software, even if advised of the possibility of such damage.
 
41
//
 
42
//M*/
 
43
 
 
44
#include "../test_precomp.hpp"
 
45
#include "opencv2/ts/ocl_test.hpp"
 
46
 
 
47
#ifdef HAVE_OPENCL
 
48
 
 
49
namespace cvtest {
 
50
namespace ocl {
 
51
 
 
52
/////////////////////////////////////////////////////////////////////////////////////////////////
 
53
// FarnebackOpticalFlow
 
54
namespace
 
55
{
 
56
    IMPLEMENT_PARAM_CLASS(PyrScale, double)
 
57
    IMPLEMENT_PARAM_CLASS(PolyN, int)
 
58
    CV_FLAGS(FarnebackOptFlowFlags, 0, OPTFLOW_FARNEBACK_GAUSSIAN)
 
59
    IMPLEMENT_PARAM_CLASS(UseInitFlow, bool)
 
60
}
 
61
 
 
62
PARAM_TEST_CASE(FarnebackOpticalFlow, PyrScale, PolyN, FarnebackOptFlowFlags, UseInitFlow)
 
63
{
 
64
    int numLevels;
 
65
    int winSize;
 
66
    int numIters;
 
67
    double pyrScale;
 
68
    int polyN;
 
69
    int flags;
 
70
    bool useInitFlow;
 
71
 
 
72
    virtual void SetUp()
 
73
    {
 
74
        numLevels = 5;
 
75
        winSize = 13;
 
76
        numIters = 10;
 
77
        pyrScale = GET_PARAM(0);
 
78
        polyN = GET_PARAM(1);
 
79
        flags = GET_PARAM(2);
 
80
        useInitFlow = GET_PARAM(3);
 
81
    }
 
82
};
 
83
 
 
84
OCL_TEST_P(FarnebackOpticalFlow, Mat)
 
85
{
 
86
    cv::Mat frame0 = readImage("optflow/RubberWhale1.png", cv::IMREAD_GRAYSCALE);
 
87
    ASSERT_FALSE(frame0.empty());
 
88
 
 
89
    cv::Mat frame1 = readImage("optflow/RubberWhale2.png", cv::IMREAD_GRAYSCALE);
 
90
    ASSERT_FALSE(frame1.empty());
 
91
 
 
92
    double polySigma = polyN <= 5 ? 1.1 : 1.5;
 
93
 
 
94
    cv::Mat flow; cv::UMat uflow;
 
95
    if (useInitFlow)
 
96
    {
 
97
        OCL_ON(cv::calcOpticalFlowFarneback(frame0, frame1, uflow, pyrScale, numLevels, winSize, numIters, polyN, polySigma, flags));
 
98
        uflow.copyTo(flow);
 
99
        flags |= cv::OPTFLOW_USE_INITIAL_FLOW;
 
100
    }
 
101
    OCL_OFF(cv::calcOpticalFlowFarneback(frame0, frame1, flow, pyrScale, numLevels, winSize, numIters, polyN, polySigma, flags));
 
102
    OCL_ON(cv::calcOpticalFlowFarneback(frame0, frame1, uflow, pyrScale, numLevels, winSize, numIters, polyN, polySigma, flags));
 
103
 
 
104
    EXPECT_MAT_SIMILAR(flow, uflow, 0.1);
 
105
}
 
106
 
 
107
 
 
108
OCL_INSTANTIATE_TEST_CASE_P(Video, FarnebackOpticalFlow,
 
109
                            Combine(
 
110
                                Values(PyrScale(0.3), PyrScale(0.5), PyrScale(0.8)),
 
111
                                Values(PolyN(5), PolyN(7)),
 
112
                                Values(FarnebackOptFlowFlags(0), FarnebackOptFlowFlags(cv::OPTFLOW_FARNEBACK_GAUSSIAN)),
 
113
                                Values(UseInitFlow(false), UseInitFlow(true))
 
114
                                )
 
115
                           );
 
116
 
 
117
 
 
118
} } // namespace cvtest::ocl
 
119
 
 
120
#endif // HAVE_OPENCL