modules/cudaimgproc/perf/perf_mean_shift.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) 2000-2008, Intel Corporation, all rights reserved.
  14 // Copyright (C) 2009, Willow Garage Inc., all rights reserved.
  15 // Third party copyrights are property of their respective owners.
  16 //
  17 // Redistribution and use in source and binary forms, with or without modification,
  18 // are permitted provided that the following conditions are met:
  19 //
  20 //   * Redistribution's of source code must retain the above copyright notice,
  21 //     this list of conditions and the following disclaimer.
  22 //
  23 //   * Redistribution's in binary form must reproduce the above copyright notice,
  24 //     this list of conditions and the following disclaimer in the documentation
  25 //     and/or other materials provided with the distribution.
  26 //
  27 //   * The name of the copyright holders may not be used to endorse or promote products
  28 //     derived from this software without specific prior written permission.
  29 //
  30 // This software is provided by the copyright holders and contributors "as is" and
  31 // any express or implied warranties, including, but not limited to, the implied
  32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  33 // In no event shall the Intel Corporation or contributors be liable for any direct,
  34 // indirect, incidental, special, exemplary, or consequential damages
  35 // (including, but not limited to, procurement of substitute goods or services;
  36 // loss of use, data, or profits; or business interruption) however caused
  37 // and on any theory of liability, whether in contract, strict liability,
  38 // or tort (including negligence or otherwise) arising in any way out of
  39 // the use of this software, even if advised of the possibility of such damage.
  40 //
  41 //M*/
  42
  43 #include "perf_precomp.hpp"
  44
  45 using namespace std;
  46 using namespace testing;
  47 using namespace perf;
  48
  49 //////////////////////////////////////////////////////////////////////
  50 // MeanShiftFiltering
  51
  52 DEF_PARAM_TEST_1(Image, string);
  53
  54 PERF_TEST_P(Image, MeanShiftFiltering,
  55             Values<string>("gpu/meanshift/cones.png"))
  56 {
  57     declare.time(300.0);
  58
  59     const cv::Mat img = readImage(GetParam());
  60     ASSERT_FALSE(img.empty());
  61
  62     cv::Mat rgba;
  63     cv::cvtColor(img, rgba, cv::COLOR_BGR2BGRA);
  64
  65     const int sp = 50;
  66     const int sr = 50;
  67
  68     if (PERF_RUN_CUDA())
  69     {
  70         const cv::cuda::GpuMat d_src(rgba);
  71         cv::cuda::GpuMat dst;
  72
  73         TEST_CYCLE() cv::cuda::meanShiftFiltering(d_src, dst, sp, sr);
  74
  75         CUDA_SANITY_CHECK(dst);
  76     }
  77     else
  78     {
  79         cv::Mat dst;
  80
  81         TEST_CYCLE() cv::pyrMeanShiftFiltering(img, dst, sp, sr);
  82
  83         CPU_SANITY_CHECK(dst);
  84     }
  85 }
  86
  87 //////////////////////////////////////////////////////////////////////
  88 // MeanShiftProc
  89
  90 PERF_TEST_P(Image, MeanShiftProc,
  91             Values<string>("gpu/meanshift/cones.png"))
  92 {
  93     declare.time(300.0);
  94
  95     const cv::Mat img = readImage(GetParam());
  96     ASSERT_FALSE(img.empty());
  97
  98     cv::Mat rgba;
  99     cv::cvtColor(img, rgba, cv::COLOR_BGR2BGRA);
 100
 101     const int sp = 50;
 102     const int sr = 50;
 103
 104     if (PERF_RUN_CUDA())
 105     {
 106         const cv::cuda::GpuMat d_src(rgba);
 107         cv::cuda::GpuMat dstr;
 108         cv::cuda::GpuMat dstsp;
 109
 110         TEST_CYCLE() cv::cuda::meanShiftProc(d_src, dstr, dstsp, sp, sr);
 111
 112         CUDA_SANITY_CHECK(dstr);
 113         CUDA_SANITY_CHECK(dstsp);
 114     }
 115     else
 116     {
 117         FAIL_NO_CPU();
 118     }
 119 }
 120
 121 //////////////////////////////////////////////////////////////////////
 122 // MeanShiftSegmentation
 123
 124 PERF_TEST_P(Image, MeanShiftSegmentation,
 125             Values<string>("gpu/meanshift/cones.png"))
 126 {
 127     declare.time(300.0);
 128
 129     const cv::Mat img = readImage(GetParam());
 130     ASSERT_FALSE(img.empty());
 131
 132     cv::Mat rgba;
 133     cv::cvtColor(img, rgba, cv::COLOR_BGR2BGRA);
 134
 135     const int sp = 10;
 136     const int sr = 10;
 137     const int minsize = 20;
 138
 139     if (PERF_RUN_CUDA())
 140     {
 141         const cv::cuda::GpuMat d_src(rgba);
 142         cv::Mat dst;
 143
 144         TEST_CYCLE() cv::cuda::meanShiftSegmentation(d_src, dst, sp, sr, minsize);
 145
 146         CUDA_SANITY_CHECK(dst);
 147     }
 148     else
 149     {
 150         FAIL_NO_CPU();
 151     }
 152 }