From b884ea36c469d8501aa4016bf76cccfc3168ccd9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 1 Feb 2022 21:17:56 +0100 Subject: [PATCH] Simplify TF normalized to pixels (#6494) --- models/common.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/models/common.py b/models/common.py index 346fa37..5571633 100644 --- a/models/common.py +++ b/models/common.py @@ -446,10 +446,7 @@ class DetectMultiBackend(nn.Module): if int8: scale, zero_point = output['quantization'] y = (y.astype(np.float32) - zero_point) * scale # re-scale - y[..., 0] *= w # x - y[..., 1] *= h # y - y[..., 2] *= w # w - y[..., 3] *= h # h + y[..., :4] *= [w, h, w, h] # xywh normalized to pixels y = torch.tensor(y) if isinstance(y, np.ndarray) else y return (y, []) if val else y