From 2b2fe26b26b5108684cbbc470b470960df783d16 Mon Sep 17 00:00:00 2001 From: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com> Date: Fri, 18 Nov 2022 19:40:39 +0800 Subject: [PATCH] fix pad_param when scale_factor > 1 (#1395) --- mmdeploy/codebase/mmdet/deploy/object_detection_model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mmdeploy/codebase/mmdet/deploy/object_detection_model.py b/mmdeploy/codebase/mmdet/deploy/object_detection_model.py index 7ca8793bb..78e5ab241 100644 --- a/mmdeploy/codebase/mmdet/deploy/object_detection_model.py +++ b/mmdeploy/codebase/mmdet/deploy/object_detection_model.py @@ -222,8 +222,10 @@ class End2EndModel(BaseBackendModel): elif 'border' in img_metas[i]: pad_key = 'border' if pad_key is not None: - x_off = img_metas[i][pad_key][2] - y_off = img_metas[i][pad_key][0] + scale_factor = img_metas[i].get('scale_factor', + np.array([1., 1.])) + x_off = img_metas[i][pad_key][2] / scale_factor[1] + y_off = img_metas[i][pad_key][0] / scale_factor[0] bboxes[:, ::2] -= x_off bboxes[:, 1::2] -= y_off bboxes *= (bboxes > 0)