105 lines
2.7 KiB
C++
105 lines
2.7 KiB
C++
#if defined(__linux__) || defined(LINUX) || defined(__APPLE__) || defined(ANDROID)
|
|
|
|
#include <opencv2/imgproc/imgproc.hpp> // Gaussian Blur
|
|
#include <opencv2/core/core.hpp> // Basic OpenCV structures (cv::Mat, Scalar)
|
|
#include <opencv2/highgui/highgui.hpp> // OpenCV window I/O
|
|
#include <opencv2/features2d/features2d.hpp>
|
|
#include <opencv2/contrib/detection_based_tracker.hpp>
|
|
|
|
#include <stdio.h>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
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)
|
|
{
|
|
Detector->detectMultiScale(Image, objects, scaleFactor, minNeighbours, 0, minObjSize, maxObjSize);
|
|
}
|
|
|
|
virtual ~CascadeDetectorAdapter()
|
|
{}
|
|
|
|
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 = "../../data/lbpcascades/lbpcascade_frontalface.xml";
|
|
cv::Ptr<cv::CascadeClassifier> cascade = makePtr<cv::CascadeClassifier>(cascadeFrontalfilename);
|
|
cv::Ptr<DetectionBasedTracker::IDetector> MainDetector = makePtr<CascadeDetectorAdapter>(cascade);
|
|
|
|
cascade = makePtr<cv::CascadeClassifier>(cascadeFrontalfilename);
|
|
cv::Ptr<DetectionBasedTracker::IDetector> TrackingDetector = makePtr<CascadeDetectorAdapter>(cascade);
|
|
|
|
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;
|
|
|
|
while(true)
|
|
{
|
|
VideoStream >> ReferenceFrame;
|
|
cvtColor(ReferenceFrame, GrayFrame, COLOR_RGB2GRAY);
|
|
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);
|
|
|
|
if (waitKey(30) >= 0) break;
|
|
}
|
|
|
|
Detector.stop();
|
|
|
|
return 0;
|
|
}
|
|
|
|
#else
|
|
|
|
#include <stdio.h>
|
|
int main()
|
|
{
|
|
printf("This sample works for UNIX or ANDROID only\n");
|
|
return 0;
|
|
}
|
|
|
|
#endif
|