opencv dnn（C++） 实时人脸检测

1
2
3
4
5
6
7
8
9
#pragma once
#include <opencv2/opencv.hpp>

using namespace cv;

class QuickDemo {
public:
	void face_detection_demo();
};

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
#include "quickopencv.h"
#include <opencv2/dnn.hpp>
#include <opencv2/opencv.hpp>

using namespace cv;
using namespace std;

void QuickDemo::face_detection_demo() {
	dnn::Net net = dnn::readNetFromTensorflow("D:\opencv\opencv\sources\samples\dnn\face_detector\opencv_face_detector_uint8.pb",
		 "D:\opencv\opencv\sources\samples\dnn\face_detector\opencv_face_detector.pbtxt");
	VideoCapture capture("C:/Users/zhaoliuliu/Pictures/Camera Roll/test1.mp4");
	Mat frame;
	while (true) {
		capture.read(frame);
		if (frame.empty()) {
			break;
		}
		Mat blob = dnn::blobFromImage(frame, 1.0, Size(300, 300), Scalar(104, 177, 123), false, false);
		net.setInput(blob);// NCHW
		Mat probs = net.forward(); // 
		Mat detectionMat(probs.size[2], probs.size[3], CV_32F, probs.ptr<float>());
		// 解析结果
		for (int i = 0; i < detectionMat.rows; i++) {
			float confidence = detectionMat.at<float>(i, 2);
			if (confidence > 0.5) {
				int x1 = static_cast<int>(detectionMat.at<float>(i, 3) * frame.cols);//后面四列表示box的四个坐标位置
				int y1 = static_cast<int>(detectionMat.at<float>(i, 4) * frame.rows);//坐标值是浮点数的比率,需要转换成像素坐标
				int x2 = static_cast<int>(detectionMat.at<float>(i, 5) * frame.cols);
				int y2 = static_cast<int>(detectionMat.at<float>(i, 6) * frame.rows);
				Rect box(x1, y1, x2 - x1, y2 - y1);
				rectangle(frame, box, Scalar(0, 0, 255), 2, 8, 0);
			}
		}
		imshow("人脸检测演示", frame);
		int c = waitKey(1);
		if (c == 27) { // 退出
			break;
		}
	}
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#include <opencv2/opencv.hpp>
#include "quickopencv.h"
#include <iostream>

using namespace cv;
using namespace std;

int main(int argc, char** argv) {
	QuickDemo qd;
	qd.face_detection_demo();

	waitKey(0);
	destroyAllWindows();
	return 0;
}