diff --git a/detect.py b/detect.py index 2d1963ad6..41c364c05 100644 --- a/detect.py +++ b/detect.py @@ -94,7 +94,7 @@ def run(weights=ROOT / 'yolov5s.pt', # model.pt path(s) imgsz = check_img_size(imgsz, s=stride) # check image size # Half - half &= (pt or jit or engine) and device.type != 'cpu' # half precision only supported by PyTorch on CUDA + half &= (pt or jit or onnx or engine) and device.type != 'cpu' # FP16 supported on limited backends with CUDA if pt or jit: model.model.half() if half else model.model.float() diff --git a/tutorial.ipynb b/tutorial.ipynb index 97f9074f9..251c18d97 100644 --- a/tutorial.ipynb +++ b/tutorial.ipynb @@ -1099,4 +1099,4 @@ "outputs": [] } ] -} \ No newline at end of file +} diff --git a/val.py b/val.py index 4709f6751..843943b5f 100644 --- a/val.py +++ b/val.py @@ -137,9 +137,9 @@ def run(data, # Load model model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data) - stride, pt, jit, engine = model.stride, model.pt, model.jit, model.engine + stride, pt, jit, onnx, engine = model.stride, model.pt, model.jit, model.onnx, model.engine imgsz = check_img_size(imgsz, s=stride) # check image size - half &= (pt or jit or engine) and device.type != 'cpu' # half precision only supported by PyTorch on CUDA + half &= (pt or jit or onnx or engine) and device.type != 'cpu' # FP16 supported on limited backends with CUDA if pt or jit: model.model.half() if half else model.model.float() elif engine: