Fix scale factor in MaskTestMixin (#4366)

* FPN deprecated warning * FPN deprecated warning * Fix scale_factor is not same type as det_bboxes error
4 years ago · bb75ddde85
parent f49540738e
commit bb75ddde85
1 changed files with 3 additions and 0 deletions
--- a/mmdet/models/roi_heads/test_mixins.py
+++ b/mmdet/models/roi_heads/test_mixins.py
@ -148,6 +148,9 @@ class MaskTestMixin(object):
            if det_bboxes.shape[0] == 0:
                segm_result = [[] for _ in range(self.mask_head.num_classes)]
            else:
+                if rescale and not isinstance(scale_factor,
+                                              (float, torch.Tensor)):
+                    scale_factor = det_bboxes.new_tensor(scale_factor)
                _bboxes = (
                    det_bboxes[:, :4] *
                    scale_factor if rescale else det_bboxes)