Blame view

3rdparty/opencv-4.5.4/modules/features2d/perf/perf_feature2d.hpp 3.66 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
  #ifndef __OPENCV_PERF_FEATURE2D_HPP__
  #define __OPENCV_PERF_FEATURE2D_HPP__
  
  #include "perf_precomp.hpp"
  
  namespace opencv_test
  {
  
  /* configuration for tests of detectors/descriptors. shared between ocl and cpu tests. */
  
  // detectors/descriptors configurations to test
  #define DETECTORS_ONLY                                                                  \
      FAST_DEFAULT, FAST_20_TRUE_TYPE5_8, FAST_20_TRUE_TYPE7_12, FAST_20_TRUE_TYPE9_16,   \
      FAST_20_FALSE_TYPE5_8, FAST_20_FALSE_TYPE7_12, FAST_20_FALSE_TYPE9_16,              \
                                                                                          \
      AGAST_DEFAULT, AGAST_5_8, AGAST_7_12d, AGAST_7_12s, AGAST_OAST_9_16,                \
                                                                                          \
      MSER_DEFAULT
  
  #define DETECTORS_EXTRACTORS                                                            \
      ORB_DEFAULT, ORB_1500_13_1,                                                         \
      AKAZE_DEFAULT, AKAZE_DESCRIPTOR_KAZE,                                               \
      BRISK_DEFAULT,                                                                      \
      KAZE_DEFAULT,                                                                       \
      SIFT_DEFAULT
  
  #define CV_ENUM_EXPAND(name, ...) CV_ENUM(name, __VA_ARGS__)
  
  enum Feature2DVals { DETECTORS_ONLY, DETECTORS_EXTRACTORS };
  CV_ENUM_EXPAND(Feature2DType, DETECTORS_ONLY, DETECTORS_EXTRACTORS)
  
  typedef tuple<Feature2DType, string> Feature2DType_String_t;
  typedef perf::TestBaseWithParam<Feature2DType_String_t> feature2d;
  
  #define TEST_IMAGES testing::Values(\
      "cv/detectors_descriptors_evaluation/images_datasets/leuven/img1.png",\
      "stitching/a3.png", \
      "stitching/s2.jpg")
  
  static inline Ptr<Feature2D> getFeature2D(Feature2DType type)
  {
      switch(type) {
      case ORB_DEFAULT:
          return ORB::create();
      case ORB_1500_13_1:
          return ORB::create(1500, 1.3f, 1);
      case FAST_DEFAULT:
          return FastFeatureDetector::create();
      case FAST_20_TRUE_TYPE5_8:
          return FastFeatureDetector::create(20, true, FastFeatureDetector::TYPE_5_8);
      case FAST_20_TRUE_TYPE7_12:
          return FastFeatureDetector::create(20, true, FastFeatureDetector::TYPE_7_12);
      case FAST_20_TRUE_TYPE9_16:
          return FastFeatureDetector::create(20, true, FastFeatureDetector::TYPE_9_16);
      case FAST_20_FALSE_TYPE5_8:
          return FastFeatureDetector::create(20, false, FastFeatureDetector::TYPE_5_8);
      case FAST_20_FALSE_TYPE7_12:
          return FastFeatureDetector::create(20, false, FastFeatureDetector::TYPE_7_12);
      case FAST_20_FALSE_TYPE9_16:
          return FastFeatureDetector::create(20, false, FastFeatureDetector::TYPE_9_16);
      case AGAST_DEFAULT:
          return AgastFeatureDetector::create();
      case AGAST_5_8:
          return AgastFeatureDetector::create(70, true, AgastFeatureDetector::AGAST_5_8);
      case AGAST_7_12d:
          return AgastFeatureDetector::create(70, true, AgastFeatureDetector::AGAST_7_12d);
      case AGAST_7_12s:
          return AgastFeatureDetector::create(70, true, AgastFeatureDetector::AGAST_7_12s);
      case AGAST_OAST_9_16:
          return AgastFeatureDetector::create(70, true, AgastFeatureDetector::OAST_9_16);
      case AKAZE_DEFAULT:
          return AKAZE::create();
      case AKAZE_DESCRIPTOR_KAZE:
          return AKAZE::create(AKAZE::DESCRIPTOR_KAZE);
      case BRISK_DEFAULT:
          return BRISK::create();
      case KAZE_DEFAULT:
          return KAZE::create();
      case MSER_DEFAULT:
          return MSER::create();
      case SIFT_DEFAULT:
          return SIFT::create();
      default:
          return Ptr<Feature2D>();
      }
  }
  
  } // namespace
  
  #endif // __OPENCV_PERF_FEATURE2D_HPP__