fix bug in postprocess

open-mmlab · Oct 8, 2023 · 985226a · 985226a
1 parent 6ff5c9b
commit 985226a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/csrc/mmdeploy/codebase/mmdet/instance_segmentation.cpp b/csrc/mmdeploy/codebase/mmdet/instance_segmentation.cpp
@@ -149,7 +149,7 @@ class ResizeInstanceMask : public ResizeBBox {
       int resize_width = int(mask_width / scale_factor_[1] + 0.5);
       // skip resize if scale_factor is 1.0
       if (resize_height != mask_height || resize_width != mask_width) {
-        cv::resize(mask_mat, mask_mat, cv::Size(resize_height, resize_width), cv::INTER_LINEAR);
+        cv::resize(mask_mat, mask_mat, cv::Size(resize_width, resize_height), cv::INTER_LINEAR);
       }
       // crop masks
       mask_mat = mask_mat(cv::Range(0, img_h), cv::Range(0, img_w)).clone();

diff --git a/mmdeploy/codebase/mmdet/deploy/object_detection.py b/mmdeploy/codebase/mmdet/deploy/object_detection.py
@@ -316,7 +316,7 @@ def get_postprocess(self, *args, **kwargs) -> Dict:
             params['score_thr'] = params['rcnn']['score_thr']
             if 'mask_thr_binary' in params['rcnn']:
                 params['mask_thr_binary'] = params['rcnn']['mask_thr_binary']
-        if 'mask_thr_binary' in params:
+        if 'mask_thr_binary' or 'mask_thr' in params:
             type = 'ResizeInstanceMask'  # for instance-seg
             # resize and crop mask to origin image
             params['is_resize_mask'] = True