diff --git a/detect.py b/detect.py index 62a4e8a..c6b0dcc 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("--int8", action='store_true', help="Datatype of model int8?") + parser.add_argument("--full_int", action='store_true', help="Datatype of model int8?") args = parser.parse_args() @@ -45,10 +45,10 @@ 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, int8=args.int8) + model = EdgeTPUModel(args.model, args.names, conf_thresh=args.conf_thresh, iou_thresh=args.iou_thresh, full_int=args.full_int) input_size = model.get_image_size() - if args.int8: + if args.full_int: x = (255*np.random.random((3,*input_size))).astype(np.int8) else: x = (255*np.random.random((3,*input_size))).astype(np.uint8)