Files
ai_sdk/app/main.cpp

27 lines
697 B
C++

#include <iostream>
#include <opencv2/opencv.hpp>
#include "util/TimeCount.h"
#include "CenterFaceMnn.h"
int32_t main(int32_t argc, char** argv) {
CenterFaceMnn::GetInstance();
std::vector<FaceInfo> faces;
cv::Mat img = cv::imread("F:/33.jpg");
int32_t count = 0;
re_test:
count++;
{
USE_TIME t(USE_TIME_US, "face_detect: ");
CenterFaceMnn::GetInstance()->Detect(img, faces, 1);
}
if (count < 10) {
goto re_test;
}
for(auto& face : faces) {
cv::rectangle(img, cv::Rect(face.x1, face.y1, face.x2 - face.x1, face.y2 - face.y1), cv::Scalar(0, 0, 255), 2);
}
cv::imshow("test", img);
cv::waitKey(0);
return 0;
}