Blame view

3rdparty/opencv-4.5.4/apps/traincascade/imagestorage.h 1.18 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
  #ifndef _OPENCV_IMAGESTORAGE_H_
  #define _OPENCV_IMAGESTORAGE_H_
  
  
  class CvCascadeImageReader
  {
  public:
      bool create( const std::string _posFilename, const std::string _negFilename, cv::Size _winSize );
      void restart() { posReader.restart(); }
      bool getNeg(cv::Mat &_img) { return negReader.get( _img ); }
      bool getPos(cv::Mat &_img) { return posReader.get( _img ); }
  
  private:
      class PosReader
      {
      public:
          PosReader();
          virtual ~PosReader();
          bool create( const std::string _filename );
          bool get( cv::Mat &_img );
          void restart();
  
          short* vec;
          FILE*  file;
          int    count;
          int    vecSize;
          int    last;
          int    base;
      } posReader;
  
      class NegReader
      {
      public:
          NegReader();
          bool create( const std::string _filename, cv::Size _winSize );
          bool get( cv::Mat& _img );
          bool nextImg();
  
          cv::Mat     src, img;
          std::vector<std::string> imgFilenames;
          cv::Point   offset, point;
          float   scale;
          float   scaleFactor;
          float   stepFactor;
          size_t  last, round;
          cv::Size    winSize;
      } negReader;
  };
  
  #endif