Add Mobilenetv2 SSD Lite post and preprocessing, add mobilenet demo

Signed-off-by: xavier <micaelaverucchi@gmail.com>
This commit is contained in:
xavier
2020-02-26 17:42:20 +01:00
parent 110bf56dc4
commit 3eb079dd13
16 changed files with 1156 additions and 202 deletions
+93
View File
@@ -0,0 +1,93 @@
#include <iostream>
#include <signal.h>
#include <stdlib.h> /* srand, rand */
#include <unistd.h>
#include <mutex>
#include "utils.h"
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/videoio.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include "MobilenetDetection.h"
bool gRun;
bool SAVE_RESULT = false;
void sig_handler(int signo)
{
std::cout << "request gateway stop\n";
gRun = false;
}
int main(int argc, char *argv[])
{
std::cout << "detection\n";
signal(SIGINT, sig_handler);
char *net = "mobilenetv2ssd.rt";
if (argc > 1)
net = argv[1];
char *input = "../demo/yolo_test.mp4";
if (argc > 2)
input = argv[2];
tk::dnn::MobilenetDetection mbnet;
mbnet.init(net);
gRun = true;
cv::VideoCapture cap(input);
if (!cap.isOpened())
gRun = false;
else
std::cout << "camera started\n";
cv::VideoWriter resultVideo;
if (SAVE_RESULT)
{
int w = cap.get(cv::CAP_PROP_FRAME_WIDTH);
int h = cap.get(cv::CAP_PROP_FRAME_HEIGHT);
resultVideo.open("result.mp4", cv::VideoWriter::fourcc('M', 'P', '4', 'V'), 30, cv::Size(w, h));
}
cv::Mat frame;
cv::Mat dnn_input;
cv::namedWindow("detection", cv::WINDOW_NORMAL);
while (gRun)
{
cap >> frame;
if (!frame.data)
{
break;
}
// this will be resized to the net format
dnn_input = frame.clone();
// TODO: async infer
mbnet.update(dnn_input);
// draw dets
frame = mbnet.draw();
cv::imshow("detection", frame);
cv::waitKey(1);
if (SAVE_RESULT)
resultVideo << frame;
}
std::cout << "detection end\n";
std::cout << COL_GREENB << "\n\nTime stats:\n";
std::cout << "Min: " << *std::min_element(mbnet.stats.begin(), mbnet.stats.end()) << " ms\n";
std::cout << "Max: " << *std::max_element(mbnet.stats.begin(), mbnet.stats.end()) << " ms\n";
double mean = 0;
for (int i = 0; i < mbnet.stats.size(); i++)
mean += mbnet.stats[i];
mean /= mbnet.stats.size();
std::cout << "Avg: " << mean << " ms\n"
<< COL_END;
return 0;
}