Skip to content

Commit 9ccf2f1

Browse files
authored
1 parent 97c1dc5 commit 9ccf2f1

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

csrc/mmdeploy/codebase/mmdet/instance_segmentation.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -145,8 +145,8 @@ class ResizeInstanceMask : public ResizeBBox {
145145
OUTCOME_TRY(auto cpu_mask, MakeAvailableOnDevice(d_mask, host, stream_));
146146
OUTCOME_TRY(stream().Wait());
147147
cv::Mat mask_mat(mask_height, mask_width, CV_32FC(mask_channel), cpu_mask.data());
148-
int resize_height = int(mask_height / scale_factor_[0] + 0.5);
149-
int resize_width = int(mask_width / scale_factor_[1] + 0.5);
148+
int resize_height = int(mask_height / scale_factor_[1] + 0.5);
149+
int resize_width = int(mask_width / scale_factor_[0] + 0.5);
150150
// skip resize if scale_factor is 1.0
151151
if (resize_height != mask_height || resize_width != mask_width) {
152152
cv::resize(mask_mat, mask_mat, cv::Size(resize_width, resize_height), cv::INTER_LINEAR);

mmdeploy/codebase/mmdet/deploy/object_detection_model.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -225,8 +225,8 @@ def postprocessing_results(self,
225225
elif 'border' in img_metas[i]:
226226
pad_key = 'border'
227227
if pad_key is not None:
228-
x_off = img_metas[i][pad_key][2] / scale_factor[1]
229-
y_off = img_metas[i][pad_key][0] / scale_factor[0]
228+
x_off = img_metas[i][pad_key][2] / scale_factor[0]
229+
y_off = img_metas[i][pad_key][0] / scale_factor[1]
230230
bboxes[:, ::2] -= x_off
231231
bboxes[:, 1::2] -= y_off
232232
bboxes *= (bboxes > 0)
@@ -262,9 +262,9 @@ def postprocessing_results(self,
262262
masks.unsqueeze(0),
263263
size=[
264264
math.ceil(masks.shape[-2] /
265-
img_metas[i]['scale_factor'][0]),
265+
img_metas[i]['scale_factor'][1]),
266266
math.ceil(masks.shape[-1] /
267-
img_metas[i]['scale_factor'][1])
267+
img_metas[i]['scale_factor'][0])
268268
])[..., :ori_h, :ori_w]
269269
masks = masks.squeeze(0)
270270
if masks.dtype != bool:

0 commit comments

Comments
 (0)