mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Add flag to disable visualization and save video

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-05-11 15:08:10 +02:00
parent b8686186e4
commit 7e8dbc8296
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/demo.cpp
+12 -3
View File
@@ -34,6 +34,12 @@ int main(int argc, char *argv[]) {
int n_classes = 80; int n_classes = 80;
if(argc > 4) if(argc > 4)
n_classes = atoi(argv[4]); n_classes = atoi(argv[4]);
bool show = true;
if(argc > 5)
show = atoi(argv[5]);
if(!show)
SAVE_RESULT = true;
tk::dnn::Yolo3Detection yolo; tk::dnn::Yolo3Detection yolo;
tk::dnn::CenternetDetection cnet; tk::dnn::CenternetDetection cnet;
@@ -76,7 +82,8 @@ int main(int argc, char *argv[]) {
cv::Mat frame; cv::Mat frame;
cv::Mat dnn_input; cv::Mat dnn_input;
cv::namedWindow("detection", cv::WINDOW_NORMAL); if(show)
cv::namedWindow("detection", cv::WINDOW_NORMAL);
std::vector<tk::dnn::box> detected_bbox; std::vector<tk::dnn::box> detected_bbox;
@@ -93,8 +100,10 @@ int main(int argc, char *argv[]) {
detNN->update(dnn_input); detNN->update(dnn_input);
frame = detNN->draw(frame); frame = detNN->draw(frame);
cv::imshow("detection", frame); if(show){
cv::waitKey(1); cv::imshow("detection", frame);
cv::waitKey(1);
}
if(SAVE_RESULT) if(SAVE_RESULT)
resultVideo << frame; resultVideo << frame;
} }