diff --git a/demo/demo/demo.cpp b/demo/demo/demo.cpp index a12148f..f9a477d 100644 --- a/demo/demo/demo.cpp +++ b/demo/demo/demo.cpp @@ -47,6 +47,9 @@ int main(int argc, char *argv[]) int port = 0; if (argc > 8) port = atoi(argv[8]); + bool extyolo=false; + if(argc >9) + extyolo = atoi(argv[9]); if (n_batch < 1 || n_batch > 64) FatalError("Batch dim not supported"); @@ -122,7 +125,7 @@ int main(int argc, char *argv[]) //inference detNN->update(batch_dnn_input, n_batch); - detNN->draw(batch_frame,true); + detNN->draw(batch_frame,extyolo); if (show) {