Skip to content

Commit 339f7ae

Browse files
AllentDanlvhan028
authored andcommitted
Fix MRCNN visualization due to error condition for mask resize (#1767)
1 parent 747d1c3 commit 339f7ae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmdeploy/codebase/mmdet/deploy/object_detection_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ def forward(self,
254254
else:
255255
masks = masks[:, :img_h, :img_w]
256256
# avoid to resize masks with zero dim
257-
if rescale and masks.shape[0] != 0:
257+
if export_postprocess_mask and rescale and masks.shape[0] != 0:
258258
masks = torch.nn.functional.interpolate(
259259
masks.unsqueeze(0),
260260
size=[

0 commit comments

Comments
 (0)