From ea5e97fb9b55c9d6f443892fe202f6d3d85dc566 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 21 Aug 2020 14:44:25 -0700 Subject: [PATCH] FP16 NMS (#814) --- utils/general.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/utils/general.py b/utils/general.py index ae30864..b857d33 100755 --- a/utils/general.py +++ b/utils/general.py @@ -596,8 +596,6 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, merge=False, Returns: detections with shape: nx6 (x1, y1, x2, y2, conf, cls) """ - if prediction.dtype is torch.float16: - prediction = prediction.float() # to FP32 nc = prediction[0].shape[1] - 5 # number of classes xc = prediction[..., 4] > conf_thres # candidates