-
Notifications
You must be signed in to change notification settings - Fork 7
/
demo.cpp
104 lines (82 loc) · 3.25 KB
/
demo.cpp
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#include <iostream>
#include "OpenVINO_LFFD.h"
#include <opencv2/opencv.hpp>
#include <string>
#include <thread>
#include <chrono>
#include <cstdlib>
using namespace cv;
//Even the input size is 320x240，the performance of small objection is also pretty good .
//If neccessary, you can modify the input size as you like. And , compile it .
int main(int argc, char** argv) {
if (argc !=5)
{
std::cout << " name.exe v/i modepath imagefile/videofile scale_num" << std::endl;
return -1;
}
//eg : std::string model_path="your_path/models"
std::string model_path = argv[2];
std::string image_or_video_file = argv[3];
std::string num_scale= argv[4];
int scale_num=std::stoi(num_scale);
LFFD* face_detector = new LFFD(model_path,"CPU",scale_num);
if(argv[1]=="i"){
cv::Mat image = cv::imread(image_or_video_file);
std::vector<FaceInfo > finalBox;
std::chrono::time_point<std::chrono::system_clock> t1 = std::chrono::system_clock::now();
face_detector->detect(image, finalBox, cv::Size(320,240));
std::chrono::time_point<std::chrono::system_clock> t2 = std::chrono::system_clock::now();
std::cout << "mtcnn time:" << (float)std::chrono::duration_cast<std::chrono::microseconds>(t2 - t1).count() / 1000 << "ms" << std::endl;
for (int i = 0; i < finalBox.size(); i++) {
FaceInfo facebox = finalBox[i];
cv::Rect box = cv::Rect(facebox.x1, facebox.y1, facebox.x2 - facebox.x1, facebox.y2 - facebox.y1);
cv::rectangle(image, box, cv::Scalar(255, 0, 21), 2);
}
std::cout << finalBox.size() << std::endl;
cv::imwrite("res.jpg", image);
cv::namedWindow("MNN", CV_WINDOW_NORMAL);
cv::imshow("MNN", image);
cv::waitKey();
}
if (argv[1] == "v") {
cv::VideoCapture cap(image_or_video_file);
cv::Mat image;
int MAXNUM = 100;
float AVE_TIME = 0;
float MAX_TIME = -1;
float MIN_TIME = 99999999999999;
int count = 0;
while (count < 10000) {
cap >> image;
std::vector<FaceInfo > finalBox;
std::chrono::time_point<std::chrono::system_clock> t1 = std::chrono::system_clock::now();
face_detector->detect(image, finalBox,cv::Size(320,240));
std::chrono::time_point<std::chrono::system_clock> t2 = std::chrono::system_clock::now();
float dur = (float)std::chrono::duration_cast<std::chrono::microseconds>(t2 - t1).count() / 1000;
std::cout << "lffd time:" << dur << "ms" << std::endl;
if (count > 1)
AVE_TIME += dur;
if (MAX_TIME < dur && count>1) {
MAX_TIME = dur;
}
if (MIN_TIME > dur) {
MIN_TIME = dur;
}
for (int i = 0; i < finalBox.size(); i++) {
FaceInfo facebox = finalBox[i];
cv::Rect box = cv::Rect(facebox.x1, facebox.y1, facebox.x2 - facebox.x1, facebox.y2 - facebox.y1);
cv::rectangle(image, box, cv::Scalar(255, 0, 21), 2);
}
cv::namedWindow("MNN", CV_WINDOW_NORMAL);
cv::imshow("MNN", image);
cv::waitKey(1);
count++;
}
std::cout << "IMAGE SHAPE: " << "640x480" << std::endl;
std::cout << "MAX LOOP TIMES: " << MAXNUM << std::endl;
std::cout << "AVE TIME: " << AVE_TIME / (count - 1) << " ms" << std::endl;
std::cout << "MAX TIME: " << MAX_TIME << " ms" << std::endl;
std::cout << "MIN TIME: " << MIN_TIME << " ms" << std::endl;
}
return 0;
}