From 335221b112d8b1693c066dc86588ba3760d73a40 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Artur=20Gon=C3=A7alves?= Date: Thu, 31 Oct 2024 15:11:48 +0900 Subject: [PATCH] Fix ONNX demo postprocessing --- deploy/onnx_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy/onnx_demo.py b/deploy/onnx_demo.py index 35f2713e..bf203e8f 100644 --- a/deploy/onnx_demo.py +++ b/deploy/onnx_demo.py @@ -175,9 +175,9 @@ def inference_with_postprocessing(ort_session, bboxes = bboxes.cpu().numpy() labels = labels.cpu().numpy() + bboxes /= scale_factor bboxes -= np.array( [pad_param[1], pad_param[0], pad_param[1], pad_param[0]]) - bboxes /= scale_factor bboxes[:, 0::2] = np.clip(bboxes[:, 0::2], 0, w) bboxes[:, 1::2] = np.clip(bboxes[:, 1::2], 0, h) bboxes = bboxes.round().astype('int')