本文需安装caffe和opencv库,caffe为cpu版本
检测效果图
代码实现
#include "MTCNN.h"
#include<iostream>
#include<vector>
#include<string>
#include<fstream>
#include <sys/time.h>
int main()
{
Mat tmp,tmp1;
//VideoCapture cap("rtsp://admin:123456jy@192.168.3.67/Streaming/Channels/1");
//if (!cap.isOpened())
//cout << "fail to open camera!" << endl;
MTCNN mtcnn("./MTmodel");
struct timeval starttime={0};
struct timeval endtime={0};
Mat srcimg=imread("test.jpg");
vector<FaceInfo> faceInfo;
mtcnn.Detect(srcimg,faceInfo);
for (int i = 0; i < faceInfo.size(); i++) {
int x = (int) faceInfo[i].bbox.x1;
int y = (int) faceInfo[i].bbox.y1;
int w = (int) (faceInfo[i].bbox.x2 - faceInfo[i].bbox.x1 + 1);
int h = (int) (faceInfo[i].bbox.y2 - faceInfo[i].bbox.y1 + 1);
cv::rectangle(srcimg, cv::Rect(y, x, h, w), cv::Scalar(255, 0, 0), 2);
}
imshow("12",srcimg);
waitKey(0);
}