// // main.cpp // UltraFaceTest // // Created by vealocia on 2019/10/17. // Copyright © 2019 vealocia. All rights reserved. // #include "UltraFace.hpp" #include <iostream> #include <opencv2/opencv.hpp> int main(int argc, char **argv) { UltraFace ultraface("models/version-RFB-320.bin", "models/version-RFB-320.param", 320, 240, 1, 0.7); // config model input std::string image_file = "intput.jpg"; // input image cv::Mat frame = cv::imread(image_file); if (frame.empty()) { std::cout << "read image failed" << std::endl; return -1; } ncnn::Mat inmat = ncnn::Mat::from_pixels(frame.data, ncnn::Mat::PIXEL_BGR2RGB, frame.cols, frame.rows); std::vector<FaceInfo> face_info; ultraface.detect(inmat, face_info); for (int i = 0; i < face_info.size(); i++) { auto face = face_info[i]; cv::Point pt1(face.x1, face.y1); cv::Point pt2(face.x2, face.y2); cv::rectangle(frame, pt1, pt2, cv::Scalar(0, 255, 0), 2); } cv::imshow("UltraFace", frame); cv::waitKey(); cv::imwrite("result.jpg", frame); return 0; }