modules/softcascade/perf/perf_softcascade.cpp

   1 #include "perf_precomp.hpp"
   2 #include <opencv2/imgproc.hpp>
   3
   4 using cv::Rect;
   5 using std::tr1::get;
   6
   7
   8 using namespace cv::softcascade;
   9
  10 typedef std::tr1::tuple<std::string, std::string> fixture;
  11 typedef perf::TestBaseWithParam<fixture> detect;
  12
  13
  14 namespace {
  15
  16 void extractRacts(std::vector<Detection> objectBoxes, std::vector<Rect>& rects)
  17 {
  18     rects.clear();
  19     for (int i = 0; i < (int)objectBoxes.size(); ++i)
  20         rects.push_back(objectBoxes[i].bb());
  21 }
  22
  23 }
  24
  25 PERF_TEST_P(detect, SoftCascadeDetector,
  26     testing::Combine(testing::Values(std::string("cv/cascadeandhog/cascades/inria_caltech-17.01.2013.xml")),
  27     testing::Values(std::string("cv/cascadeandhog/images/image_00000000_0.png"))))
  28 {
  29     cv::Mat colored = cv::imread(getDataPath(get<1>(GetParam())));
  30     ASSERT_FALSE(colored.empty());
  31
  32     Detector cascade;
  33     cv::FileStorage fs(getDataPath(get<0>(GetParam())), cv::FileStorage::READ);
  34     ASSERT_TRUE(fs.isOpened());
  35     ASSERT_TRUE(cascade.load(fs.getFirstTopLevelNode()));
  36
  37     std::vector<Detection> objectBoxes;
  38     TEST_CYCLE()
  39     {
  40         cascade.detect(colored, cv::noArray(), objectBoxes);
  41     }
  42
  43     std::vector<Rect> rects;
  44     extractRacts(objectBoxes, rects);
  45     std::sort(rects.begin(), rects.end(), perf::comparators::RectLess());
  46     SANITY_CHECK(rects);
  47 }