前言
这里我使用的Opencv4.5版本,其中是要使用连个文件:1、opencv_face_detector_uint8;2、opencv_face_detector.pbtxt,通过这个链接下载,免费不用币:https://download.csdn.net/download/Galen_xia/22003259
代码实例
void face_detector_demo() {
std::string root_dir = "D:/opencv-4.5.1-vc14_vc15/opencv/sources/samples/dnn/face_detector/";
dnn::Net net = dnn::readNetFromTensorflow(root_dir + "opencv_face_detector_uint8.pb",root_dir+"opencv_face_detector.pbtxt");
VideoCapture capture("D:/opencv/1.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 res(Size(probs.size[2], probs.size[3]), CV_32F, probs.ptr<float>());
// 解析结果
for (int i = 0; i < res.rows;i++) {
float confidence = res.at<float>(i, 2);
if (confidence > 0.5) {
int x1 = static_cast<int>(res.at<float>(i, 3)*frame.cols);
int y1 = static_cast<int>(res.at<float>(i, 4)*frame.rows);
int x2 = static_cast<int>(res.at<float>(i, 5)*frame.cols);
int y2 = static_cast<int>(res.at<float>(i, 6)*frame.rows);
Rect box(x1, y1, x2 - x1, y2 - y1);
rectangle(frame, box, Scalar(0, 255, 0), 2, 8, 0);
}
}
Mat dst;
imshow("人脸检测演示:", frame);
int c = waitKey(10);
if (c == 27) {
break;
}
}
}