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

added save flag

Browse Source
This commit is contained in:
Biparnak Roy
2020-07-18 05:17:07 +05:30
committed by GitHub
parent 78944d8ce8
commit 44a6e8cca5
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/demo.cpp
+5 -6
View File
@@ -42,15 +42,15 @@ int main(int argc, char *argv[])
bool show = true; bool show = true;
if (argc > 6) if (argc > 6)
show = atoi(argv[6]); show = atoi(argv[6]);
int port = 0;
if (argc > 7) if (argc > 7)
port = atoi(argv[7]); SAVE_RESULT = atoi(argv[7]);
int port = 0;
if (argc > 8)
port = atoi(argv[8]);
if (n_batch < 1 || n_batch > 64) if (n_batch < 1 || n_batch > 64)
FatalError("Batch dim not supported"); FatalError("Batch dim not supported");
if (!show)
SAVE_RESULT = true;
tk::dnn::Yolo3Detection yolo; tk::dnn::Yolo3Detection yolo;
tk::dnn::CenternetDetection cnet; tk::dnn::CenternetDetection cnet;
@@ -129,7 +129,6 @@ int main(int argc, char *argv[])
for (int bi = 0; bi < n_batch; ++bi) for (int bi = 0; bi < n_batch; ++bi)
{ {
cv::imshow("detection", batch_frame[bi]); cv::imshow("detection", batch_frame[bi]);
} }
} }
if (cv::waitKey(1) == 27) if (cv::waitKey(1) == 27)
@@ -138,7 +137,7 @@ int main(int argc, char *argv[])
} }
if (n_batch == 1 && SAVE_RESULT) if (n_batch == 1 && SAVE_RESULT)
resultVideo << frame; resultVideo << frame;
if (port > 0) if (port > 0)
{ {
send_mjpeg(batch_frame[0], port, 400000, 40); send_mjpeg(batch_frame[0], port, 400000, 40);