update coco bench config

sebastianfis · Aug 31, 2021 · dfd979a · dfd979a
1 parent 9ed60aa
commit dfd979a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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)