From 47b1b6a006dd19840c2172f75d4bd44901f7313c Mon Sep 17 00:00:00 2001
From: Biparnak Roy <54219771+biparnakroy@users.noreply.github.com>
Date: Sat, 25 Jul 2020 22:01:01 +0530
Subject: [PATCH] added extyolo flag

---
 demo/demo/demo.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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)
         {