Fix MRCNN visualization due to error condition for mask resize (#1767)

open-mmlab · Mar 1, 2023 · 339f7ae · 339f7ae
1 parent 747d1c3
commit 339f7ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdeploy/codebase/mmdet/deploy/object_detection_model.py b/mmdeploy/codebase/mmdet/deploy/object_detection_model.py
@@ -254,7 +254,7 @@ def forward(self,
                 else:
                     masks = masks[:, :img_h, :img_w]
                 # avoid to resize masks with zero dim
-                if rescale and masks.shape[0] != 0:
+                if export_postprocess_mask and rescale and masks.shape[0] != 0:
                     masks = torch.nn.functional.interpolate(
                         masks.unsqueeze(0),
                         size=[