modules/imgproc/perf/perf_cvt_color.cpp

   1 #include "perf_precomp.hpp"
   2
   3 using namespace std;
   4 using namespace cv;
   5 using namespace perf;
   6 using std::tr1::make_tuple;
   7 using std::tr1::get;
   8
   9 //extra color conversions supported implicitly
  10 enum
  11 {
  12     CX_BGRA2HLS      = CV_COLORCVT_MAX + CV_BGR2HLS,
  13     CX_BGRA2HLS_FULL = CV_COLORCVT_MAX + CV_BGR2HLS_FULL,
  14     CX_BGRA2HSV      = CV_COLORCVT_MAX + CV_BGR2HSV,
  15     CX_BGRA2HSV_FULL = CV_COLORCVT_MAX + CV_BGR2HSV_FULL,
  16     CX_BGRA2Lab      = CV_COLORCVT_MAX + CV_BGR2Lab,
  17     CX_BGRA2Luv      = CV_COLORCVT_MAX + CV_BGR2Luv,
  18     CX_BGRA2XYZ      = CV_COLORCVT_MAX + CV_BGR2XYZ,
  19     CX_BGRA2YCrCb    = CV_COLORCVT_MAX + CV_BGR2YCrCb,
  20     CX_BGRA2YUV      = CV_COLORCVT_MAX + CV_BGR2YUV,
  21     CX_HLS2BGRA      = CV_COLORCVT_MAX + CV_HLS2BGR,
  22     CX_HLS2BGRA_FULL = CV_COLORCVT_MAX + CV_HLS2BGR_FULL,
  23     CX_HLS2RGBA      = CV_COLORCVT_MAX + CV_HLS2RGB,
  24     CX_HLS2RGBA_FULL = CV_COLORCVT_MAX + CV_HLS2RGB_FULL,
  25     CX_HSV2BGRA      = CV_COLORCVT_MAX + CV_HSV2BGR,
  26     CX_HSV2BGRA_FULL = CV_COLORCVT_MAX + CV_HSV2BGR_FULL,
  27     CX_HSV2RGBA      = CV_COLORCVT_MAX + CV_HSV2RGB,
  28     CX_HSV2RGBA_FULL = CV_COLORCVT_MAX + CV_HSV2RGB_FULL,
  29     CX_Lab2BGRA      = CV_COLORCVT_MAX + CV_Lab2BGR,
  30     CX_Lab2LBGRA     = CV_COLORCVT_MAX + CV_Lab2LBGR,
  31     CX_Lab2LRGBA     = CV_COLORCVT_MAX + CV_Lab2LRGB,
  32     CX_Lab2RGBA      = CV_COLORCVT_MAX + CV_Lab2RGB,
  33     CX_LBGRA2Lab     = CV_COLORCVT_MAX + CV_LBGR2Lab,
  34     CX_LBGRA2Luv     = CV_COLORCVT_MAX + CV_LBGR2Luv,
  35     CX_LRGBA2Lab     = CV_COLORCVT_MAX + CV_LRGB2Lab,
  36     CX_LRGBA2Luv     = CV_COLORCVT_MAX + CV_LRGB2Luv,
  37     CX_Luv2BGRA      = CV_COLORCVT_MAX + CV_Luv2BGR,
  38     CX_Luv2LBGRA     = CV_COLORCVT_MAX + CV_Luv2LBGR,
  39     CX_Luv2LRGBA     = CV_COLORCVT_MAX + CV_Luv2LRGB,
  40     CX_Luv2RGBA      = CV_COLORCVT_MAX + CV_Luv2RGB,
  41     CX_RGBA2HLS      = CV_COLORCVT_MAX + CV_RGB2HLS,
  42     CX_RGBA2HLS_FULL = CV_COLORCVT_MAX + CV_RGB2HLS_FULL,
  43     CX_RGBA2HSV      = CV_COLORCVT_MAX + CV_RGB2HSV,
  44     CX_RGBA2HSV_FULL = CV_COLORCVT_MAX + CV_RGB2HSV_FULL,
  45     CX_RGBA2Lab      = CV_COLORCVT_MAX + CV_RGB2Lab,
  46     CX_RGBA2Luv      = CV_COLORCVT_MAX + CV_RGB2Luv,
  47     CX_RGBA2XYZ      = CV_COLORCVT_MAX + CV_RGB2XYZ,
  48     CX_RGBA2YCrCb    = CV_COLORCVT_MAX + CV_RGB2YCrCb,
  49     CX_RGBA2YUV      = CV_COLORCVT_MAX + CV_RGB2YUV,
  50     CX_XYZ2BGRA      = CV_COLORCVT_MAX + CV_XYZ2BGR,
  51     CX_XYZ2RGBA      = CV_COLORCVT_MAX + CV_XYZ2RGB,
  52     CX_YCrCb2BGRA    = CV_COLORCVT_MAX + CV_YCrCb2BGR,
  53     CX_YCrCb2RGBA    = CV_COLORCVT_MAX + CV_YCrCb2RGB,
  54     CX_YUV2BGRA      = CV_COLORCVT_MAX + CV_YUV2BGR,
  55     CX_YUV2RGBA      = CV_COLORCVT_MAX + CV_YUV2RGB
  56 };
  57
  58 CV_ENUM(CvtMode,
  59     CV_BayerBG2BGR, CV_BayerBG2BGR_VNG, CV_BayerBG2GRAY,
  60     CV_BayerGB2BGR, CV_BayerGB2BGR_VNG, CV_BayerGB2GRAY,
  61     CV_BayerGR2BGR, CV_BayerGR2BGR_VNG, CV_BayerGR2GRAY,
  62     CV_BayerRG2BGR, CV_BayerRG2BGR_VNG, CV_BayerRG2GRAY,
  63
  64     CV_BGR2BGR555, CV_BGR2BGR565, CV_BGR2BGRA, CV_BGR2GRAY,
  65     CV_BGR2HLS, CV_BGR2HLS_FULL, CV_BGR2HSV, CV_BGR2HSV_FULL,
  66     CV_BGR2Lab, CV_BGR2Luv, CV_BGR2RGB, CV_BGR2RGBA, CV_BGR2XYZ,
  67     CV_BGR2YCrCb, CV_BGR2YUV, CV_BGR5552BGR, CV_BGR5552BGRA,
  68
  69     CV_BGR5552GRAY, CV_BGR5552RGB, CV_BGR5552RGBA, CV_BGR5652BGR,
  70     CV_BGR5652BGRA, CV_BGR5652GRAY, CV_BGR5652RGB, CV_BGR5652RGBA,
  71
  72     CV_BGRA2BGR, CV_BGRA2BGR555, CV_BGRA2BGR565, CV_BGRA2GRAY, CV_BGRA2RGBA,
  73     CX_BGRA2HLS, CX_BGRA2HLS_FULL, CX_BGRA2HSV, CX_BGRA2HSV_FULL,
  74     CX_BGRA2Lab, CX_BGRA2Luv, CX_BGRA2XYZ,
  75     CX_BGRA2YCrCb, CX_BGRA2YUV,
  76
  77     CV_GRAY2BGR, CV_GRAY2BGR555, CV_GRAY2BGR565, CV_GRAY2BGRA,
  78
  79     CV_HLS2BGR, CV_HLS2BGR_FULL, CV_HLS2RGB, CV_HLS2RGB_FULL,
  80     CX_HLS2BGRA, CX_HLS2BGRA_FULL, CX_HLS2RGBA, CX_HLS2RGBA_FULL,
  81
  82     CV_HSV2BGR, CV_HSV2BGR_FULL, CV_HSV2RGB, CV_HSV2RGB_FULL,
  83     CX_HSV2BGRA, CX_HSV2BGRA_FULL, CX_HSV2RGBA,    CX_HSV2RGBA_FULL,
  84
  85     CV_Lab2BGR, CV_Lab2LBGR, CV_Lab2LRGB, CV_Lab2RGB,
  86     CX_Lab2BGRA, CX_Lab2LBGRA, CX_Lab2LRGBA, CX_Lab2RGBA,
  87
  88     CV_LBGR2Lab, CV_LBGR2Luv, CV_LRGB2Lab, CV_LRGB2Luv,
  89     CX_LBGRA2Lab, CX_LBGRA2Luv, CX_LRGBA2Lab, CX_LRGBA2Luv,
  90
  91     CV_Luv2BGR, CV_Luv2LBGR, CV_Luv2LRGB, CV_Luv2RGB,
  92     CX_Luv2BGRA, CX_Luv2LBGRA, CX_Luv2LRGBA, CX_Luv2RGBA,
  93
  94     CV_RGB2BGR555, CV_RGB2BGR565, CV_RGB2GRAY,
  95     CV_RGB2HLS, CV_RGB2HLS_FULL, CV_RGB2HSV, CV_RGB2HSV_FULL,
  96     CV_RGB2Lab, CV_RGB2Luv, CV_RGB2XYZ, CV_RGB2YCrCb, CV_RGB2YUV,
  97
  98     CV_RGBA2BGR, CV_RGBA2BGR555, CV_RGBA2BGR565, CV_RGBA2GRAY,
  99     CX_RGBA2HLS, CX_RGBA2HLS_FULL, CX_RGBA2HSV, CX_RGBA2HSV_FULL,
 100     CX_RGBA2Lab, CX_RGBA2Luv, CX_RGBA2XYZ,
 101     CX_RGBA2YCrCb, CX_RGBA2YUV,
 102
 103     CV_XYZ2BGR, CV_XYZ2RGB, CX_XYZ2BGRA, CX_XYZ2RGBA,
 104
 105     CV_YCrCb2BGR, CV_YCrCb2RGB, CX_YCrCb2BGRA, CX_YCrCb2RGBA,
 106     CV_YUV2BGR, CV_YUV2RGB, CX_YUV2BGRA, CX_YUV2RGBA
 107     )
 108
 109 CV_ENUM(CvtMode2, CV_YUV2BGR_NV12, CV_YUV2BGRA_NV12, CV_YUV2RGB_NV12, CV_YUV2RGBA_NV12, CV_YUV2BGR_NV21, CV_YUV2BGRA_NV21, CV_YUV2RGB_NV21, CV_YUV2RGBA_NV21,
 110                   CV_YUV2BGR_YV12, CV_YUV2BGRA_YV12, CV_YUV2RGB_YV12, CV_YUV2RGBA_YV12, CV_YUV2BGR_IYUV, CV_YUV2BGRA_IYUV, CV_YUV2RGB_IYUV, CV_YUV2RGBA_IYUV,
 111                   COLOR_YUV2GRAY_420, CV_YUV2RGB_UYVY, CV_YUV2BGR_UYVY, CV_YUV2RGBA_UYVY, CV_YUV2BGRA_UYVY, CV_YUV2RGB_YUY2, CV_YUV2BGR_YUY2, CV_YUV2RGB_YVYU,
 112                   CV_YUV2BGR_YVYU, CV_YUV2RGBA_YUY2, CV_YUV2BGRA_YUY2, CV_YUV2RGBA_YVYU, CV_YUV2BGRA_YVYU)
 113
 114 struct ChPair
 115 {
 116     ChPair(int _scn, int _dcn): scn(_scn), dcn(_dcn) {}
 117     int scn, dcn;
 118 };
 119
 120 ChPair getConversionInfo(int cvtMode)
 121 {
 122     switch(cvtMode)
 123     {
 124     case CV_BayerBG2GRAY: case CV_BayerGB2GRAY:
 125     case CV_BayerGR2GRAY: case CV_BayerRG2GRAY:
 126     case CV_YUV2GRAY_420:
 127         return ChPair(1,1);
 128     case CV_GRAY2BGR555: case CV_GRAY2BGR565:
 129         return ChPair(1,2);
 130     case CV_BayerBG2BGR: case CV_BayerBG2BGR_VNG:
 131     case CV_BayerGB2BGR: case CV_BayerGB2BGR_VNG:
 132     case CV_BayerGR2BGR: case CV_BayerGR2BGR_VNG:
 133     case CV_BayerRG2BGR: case CV_BayerRG2BGR_VNG:
 134     case CV_GRAY2BGR:
 135     case CV_YUV2BGR_NV12: case CV_YUV2RGB_NV12:
 136     case CV_YUV2BGR_NV21: case CV_YUV2RGB_NV21:
 137     case CV_YUV2BGR_YV12: case CV_YUV2RGB_YV12:
 138     case CV_YUV2BGR_IYUV: case CV_YUV2RGB_IYUV:
 139         return ChPair(1,3);
 140     case CV_GRAY2BGRA:
 141     case CV_YUV2BGRA_NV12: case CV_YUV2RGBA_NV12:
 142     case CV_YUV2BGRA_NV21: case CV_YUV2RGBA_NV21:
 143     case CV_YUV2BGRA_YV12: case CV_YUV2RGBA_YV12:
 144     case CV_YUV2BGRA_IYUV: case CV_YUV2RGBA_IYUV:
 145         return ChPair(1,4);
 146     case CV_BGR5552GRAY: case CV_BGR5652GRAY:
 147         return ChPair(2,1);
 148     case CV_BGR5552BGR: case CV_BGR5552RGB:
 149     case CV_BGR5652BGR: case CV_BGR5652RGB:
 150     case CV_YUV2RGB_UYVY: case CV_YUV2BGR_UYVY:
 151     case CV_YUV2RGBA_UYVY: case CV_YUV2BGRA_UYVY:
 152     case CV_YUV2RGB_YUY2: case CV_YUV2BGR_YUY2:
 153     case CV_YUV2RGB_YVYU: case CV_YUV2BGR_YVYU:
 154     case CV_YUV2RGBA_YUY2: case CV_YUV2BGRA_YUY2:
 155     case CV_YUV2RGBA_YVYU: case CV_YUV2BGRA_YVYU:
 156         return ChPair(2,3);
 157     case CV_BGR5552BGRA: case CV_BGR5552RGBA:
 158     case CV_BGR5652BGRA: case CV_BGR5652RGBA:
 159         return ChPair(2,4);
 160     case CV_BGR2GRAY: case CV_RGB2GRAY:
 161         return ChPair(3,1);
 162     case CV_BGR2BGR555: case CV_BGR2BGR565:
 163     case CV_RGB2BGR555: case CV_RGB2BGR565:
 164         return ChPair(3,2);
 165     case CV_BGR2HLS: case CV_BGR2HLS_FULL:
 166     case CV_BGR2HSV: case CV_BGR2HSV_FULL:
 167     case CV_BGR2Lab: case CV_BGR2Luv:
 168     case CV_BGR2RGB: case CV_BGR2XYZ:
 169     case CV_BGR2YCrCb: case CV_BGR2YUV:
 170     case CV_HLS2BGR: case CV_HLS2BGR_FULL:
 171     case CV_HLS2RGB: case CV_HLS2RGB_FULL:
 172     case CV_HSV2BGR: case CV_HSV2BGR_FULL:
 173     case CV_HSV2RGB: case CV_HSV2RGB_FULL:
 174     case CV_Lab2BGR: case CV_Lab2LBGR:
 175     case CV_Lab2LRGB: case CV_Lab2RGB:
 176     case CV_LBGR2Lab: case CV_LBGR2Luv:
 177     case CV_LRGB2Lab: case CV_LRGB2Luv:
 178     case CV_Luv2BGR: case CV_Luv2LBGR:
 179     case CV_Luv2LRGB: case CV_Luv2RGB:
 180     case CV_RGB2HLS: case CV_RGB2HLS_FULL:
 181     case CV_RGB2HSV: case CV_RGB2HSV_FULL:
 182     case CV_RGB2Lab: case CV_RGB2Luv:
 183     case CV_RGB2XYZ: case CV_RGB2YCrCb:
 184     case CV_RGB2YUV: case CV_XYZ2BGR:
 185     case CV_XYZ2RGB: case CV_YCrCb2BGR:
 186     case CV_YCrCb2RGB: case CV_YUV2BGR:
 187     case CV_YUV2RGB:
 188         return ChPair(3,3);
 189     case CV_BGR2BGRA: case CV_BGR2RGBA:
 190     case CX_HLS2BGRA: case CX_HLS2BGRA_FULL:
 191     case CX_HLS2RGBA: case CX_HLS2RGBA_FULL:
 192     case CX_HSV2BGRA: case CX_HSV2BGRA_FULL:
 193     case CX_HSV2RGBA: case CX_HSV2RGBA_FULL:
 194     case CX_Lab2BGRA: case CX_Lab2LBGRA:
 195     case CX_Lab2LRGBA: case CX_Lab2RGBA:
 196     case CX_Luv2BGRA: case CX_Luv2LBGRA:
 197     case CX_Luv2LRGBA: case CX_Luv2RGBA:
 198     case CX_XYZ2BGRA: case CX_XYZ2RGBA:
 199     case CX_YCrCb2BGRA: case CX_YCrCb2RGBA:
 200     case CX_YUV2BGRA: case CX_YUV2RGBA:
 201         return ChPair(3,4);
 202     case CV_BGRA2GRAY: case CV_RGBA2GRAY:
 203         return ChPair(4,1);
 204     case CV_BGRA2BGR555: case CV_BGRA2BGR565:
 205     case CV_RGBA2BGR555: case CV_RGBA2BGR565:
 206         return ChPair(4,2);
 207     case CV_BGRA2BGR: case CX_BGRA2HLS:
 208     case CX_BGRA2HLS_FULL: case CX_BGRA2HSV:
 209     case CX_BGRA2HSV_FULL: case CX_BGRA2Lab:
 210     case CX_BGRA2Luv: case CX_BGRA2XYZ:
 211     case CX_BGRA2YCrCb: case CX_BGRA2YUV:
 212     case CX_LBGRA2Lab: case CX_LBGRA2Luv:
 213     case CX_LRGBA2Lab: case CX_LRGBA2Luv:
 214     case CV_RGBA2BGR: case CX_RGBA2HLS:
 215     case CX_RGBA2HLS_FULL: case CX_RGBA2HSV:
 216     case CX_RGBA2HSV_FULL: case CX_RGBA2Lab:
 217     case CX_RGBA2Luv: case CX_RGBA2XYZ:
 218     case CX_RGBA2YCrCb: case CX_RGBA2YUV:
 219         return ChPair(4,3);
 220     case CV_BGRA2RGBA:
 221         return ChPair(4,4);
 222     default:
 223         ADD_FAILURE() << "Unknown conversion type";
 224         break;
 225     };
 226     return ChPair(0,0);
 227 }
 228
 229 typedef std::tr1::tuple<Size, CvtMode> Size_CvtMode_t;
 230 typedef perf::TestBaseWithParam<Size_CvtMode_t> Size_CvtMode;
 231
 232 PERF_TEST_P(Size_CvtMode, cvtColor8u,
 233             testing::Combine(
 234                 testing::Values(TYPICAL_MAT_SIZES),
 235                 testing::ValuesIn(CvtMode::all())
 236                 )
 237             )
 238 {
 239     Size sz = get<0>(GetParam());
 240     int mode = get<1>(GetParam());
 241     ChPair ch = getConversionInfo(mode);
 242     mode %= CV_COLORCVT_MAX;
 243
 244     Mat src(sz, CV_8UC(ch.scn));
 245     Mat dst(sz, CV_8UC(ch.dcn));
 246
 247     declare.time(100);
 248     declare.in(src, WARMUP_RNG).out(dst);
 249
 250     TEST_CYCLE() cvtColor(src, dst, mode, ch.dcn);
 251
 252     SANITY_CHECK(dst, 1);
 253 }
 254
 255 typedef std::tr1::tuple<Size, CvtMode2> Size_CvtMode2_t;
 256 typedef perf::TestBaseWithParam<Size_CvtMode2_t> Size_CvtMode2;
 257
 258 PERF_TEST_P(Size_CvtMode2, cvtColorYUV420,
 259             testing::Combine(
 260                 testing::Values(szVGA, sz720p, sz1080p, Size(130, 60)),
 261                 testing::ValuesIn(CvtMode2::all())
 262                 )
 263             )
 264 {
 265     Size sz = get<0>(GetParam());
 266     int mode = get<1>(GetParam());
 267     ChPair ch = getConversionInfo(mode);
 268
 269     Mat src(sz.height + sz.height / 2, sz.width, CV_8UC(ch.scn));
 270     Mat dst(sz, CV_8UC(ch.dcn));
 271
 272     declare.time(100);
 273     declare.in(src, WARMUP_RNG).out(dst);
 274
 275     TEST_CYCLE() cvtColor(src, dst, mode, ch.dcn);
 276
 277     SANITY_CHECK(dst, 1);
 278 }
 279
 280 CV_ENUM(EdgeAwareBayerMode, COLOR_BayerBG2BGR_EA, COLOR_BayerGB2BGR_EA, COLOR_BayerRG2BGR_EA, COLOR_BayerGR2BGR_EA)
 281
 282 typedef std::tr1::tuple<Size, EdgeAwareBayerMode> EdgeAwareParams;
 283 typedef perf::TestBaseWithParam<EdgeAwareParams> EdgeAwareDemosaicingTest;
 284
 285 PERF_TEST_P(EdgeAwareDemosaicingTest, demosaicingEA,
 286             testing::Combine(
 287                 testing::Values(szVGA, sz720p, sz1080p, Size(130, 60)),
 288                 testing::ValuesIn(EdgeAwareBayerMode::all())
 289                 )
 290             )
 291 {
 292     Size sz = get<0>(GetParam());
 293     int mode = get<1>(GetParam());
 294
 295     Mat src(sz, CV_8UC1);
 296     Mat dst(sz, CV_8UC3);
 297
 298     declare.in(src, WARMUP_RNG).out(dst);
 299
 300     TEST_CYCLE() cvtColor(src, dst, mode, 3);
 301
 302     SANITY_CHECK(dst, 1);
 303 }