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')