Blame view

3rdparty/opencv-4.5.4/samples/cpp/dbt_face_detection.cpp 2.96 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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
  #if defined(__linux__) || defined(LINUX) || defined(__APPLE__) || defined(ANDROID) || (defined(_MSC_VER) && _MSC_VER>=1800)
  
  #include <opencv2/imgproc.hpp>  // Gaussian Blur
  #include <opencv2/core.hpp>        // Basic OpenCV structures (cv::Mat, Scalar)
  #include <opencv2/videoio.hpp>
  #include <opencv2/highgui.hpp>  // OpenCV window I/O
  #include <opencv2/features2d.hpp>
  #include <opencv2/objdetect.hpp>
  
  #include <stdio.h>
  
  using namespace std;
  using namespace cv;
  
  const string WindowName = "Face Detection example";
  
  class CascadeDetectorAdapter: public DetectionBasedTracker::IDetector
  {
      public:
          CascadeDetectorAdapter(cv::Ptr<cv::CascadeClassifier> detector):
              IDetector(),
              Detector(detector)
          {
              CV_Assert(detector);
          }
  
          void detect(const cv::Mat &Image, std::vector<cv::Rect> &objects) CV_OVERRIDE
          {
              Detector->detectMultiScale(Image, objects, scaleFactor, minNeighbours, 0, minObjSize, maxObjSize);
          }
  
          virtual ~CascadeDetectorAdapter() CV_OVERRIDE
          {}
  
      private:
          CascadeDetectorAdapter();
          cv::Ptr<cv::CascadeClassifier> Detector;
   };
  
  int main(int , char** )
  {
      namedWindow(WindowName);
  
      VideoCapture VideoStream(0);
  
      if (!VideoStream.isOpened())
      {
          printf("Error: Cannot open video stream from camera\n");
          return 1;
      }
  
      std::string cascadeFrontalfilename = samples::findFile("data/lbpcascades/lbpcascade_frontalface.xml");
      cv::Ptr<cv::CascadeClassifier> cascade = makePtr<cv::CascadeClassifier>(cascadeFrontalfilename);
      cv::Ptr<DetectionBasedTracker::IDetector> MainDetector = makePtr<CascadeDetectorAdapter>(cascade);
      if ( cascade->empty() )
      {
        printf("Error: Cannot load %s\n", cascadeFrontalfilename.c_str());
        return 2;
      }
  
      cascade = makePtr<cv::CascadeClassifier>(cascadeFrontalfilename);
      cv::Ptr<DetectionBasedTracker::IDetector> TrackingDetector = makePtr<CascadeDetectorAdapter>(cascade);
      if ( cascade->empty() )
      {
        printf("Error: Cannot load %s\n", cascadeFrontalfilename.c_str());
        return 2;
      }
  
      DetectionBasedTracker::Parameters params;
      DetectionBasedTracker Detector(MainDetector, TrackingDetector, params);
  
      if (!Detector.run())
      {
          printf("Error: Detector initialization failed\n");
          return 2;
      }
  
      Mat ReferenceFrame;
      Mat GrayFrame;
      vector<Rect> Faces;
  
      do
      {
          VideoStream >> ReferenceFrame;
          cvtColor(ReferenceFrame, GrayFrame, COLOR_BGR2GRAY);
          Detector.process(GrayFrame);
          Detector.getObjects(Faces);
  
          for (size_t i = 0; i < Faces.size(); i++)
          {
              rectangle(ReferenceFrame, Faces[i], Scalar(0,255,0));
          }
  
          imshow(WindowName, ReferenceFrame);
      } while (waitKey(30) < 0);
  
      Detector.stop();
  
      return 0;
  }
  
  #else
  
  #include <stdio.h>
  int main()
  {
      printf("This sample works for UNIX or ANDROID or Visual Studio 2013+ only\n");
      return 0;
  }
  
  #endif