From dfd979a63e1e196bae44d6949a9695352e012b67 Mon Sep 17 00:00:00 2001
From: Josh <jveitchmichaelis@gmail.com>
Date: Wed, 1 Sep 2021 09:10:46 +1200
Subject: [PATCH] update coco bench config

---
 test_edgetpu.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test_edgetpu.py b/test_edgetpu.py
index 051959e..5859a0d 100644
--- a/test_edgetpu.py
+++ b/test_edgetpu.py
@@ -31,7 +31,7 @@
     parser.add_argument("--bench_coco", action='store_true', help="Process a stream")
     parser.add_argument("--coco_path", type=str, help="Path to COCO 2017 Val folder")
     parser.add_argument("--quiet","-q", action='store_true', help="Disable logging (except errors)")
-    
+        
     args = parser.parse_args()
     
     if args.quiet:
@@ -89,6 +89,9 @@
     elif args.bench_coco:
         logger.info("Testing on COCO dataset")
         
+        model.conf_thresh = 0.001
+        model.iou_thresh = 0.65
+        
         coco_glob = os.path.join(args.coco_path, "*.jpg")
         images = glob.glob(coco_glob)