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

added extyolo flag

Browse Source
This commit is contained in:
Biparnak Roy
2020-07-25 22:01:01 +05:30
committed by GitHub
parent 07fe1c8fdb
commit 47b1b6a006
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/demo.cpp
+4 -1
View File
@@ -47,6 +47,9 @@ int main(int argc, char *argv[])
int port = 0; int port = 0;
if (argc > 8) if (argc > 8)
port = atoi(argv[8]); port = atoi(argv[8]);
bool extyolo=false;
if(argc >9)
extyolo = atoi(argv[9]);
if (n_batch < 1 || n_batch > 64) if (n_batch < 1 || n_batch > 64)
FatalError("Batch dim not supported"); FatalError("Batch dim not supported");
@@ -122,7 +125,7 @@ int main(int argc, char *argv[])
//inference //inference
detNN->update(batch_dnn_input, n_batch); detNN->update(batch_dnn_input, n_batch);
detNN->draw(batch_frame,true); detNN->draw(batch_frame,extyolo);
if (show) if (show)
{ {