From 1f53bca5213e55a408101d67c5119a76ded6d1f3 Mon Sep 17 00:00:00 2001 From: sebastianfis <100689589+sebastianfis@users.noreply.github.com> Date: Sat, 23 Mar 2024 22:29:05 +0100 Subject: [PATCH] Update detect.py --- detect.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/detect.py b/detect.py index c6b0dcc..bfc7a62 100644 --- a/detect.py +++ b/detect.py @@ -33,7 +33,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)") - parser.add_argument("--full_int", action='store_true', help="Datatype of model int8?") + parser.add_argument("--v8", action='store_true', help="yolov8 model?") args = parser.parse_args() @@ -45,7 +45,7 @@ logger.error("Please select either an input image or a stream") exit(1) - model = EdgeTPUModel(args.model, args.names, conf_thresh=args.conf_thresh, iou_thresh=args.iou_thresh, full_int=args.full_int) + model = EdgeTPUModel(args.model, args.names, conf_thresh=args.conf_thresh, iou_thresh=args.iou_thresh, full_int=args.v8) input_size = model.get_image_size() if args.full_int: