Update D2Det.py

JialeCao001 · Jul 5, 2020 · 1814162 · 1814162
1 parent 89b6279
commit 1814162
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/models/detectors/D2Det.py b/mmdet/models/detectors/D2Det.py
@@ -277,7 +277,7 @@ def simple_test(self, img, img_meta, proposals=None, rescale=False):
                 mask_scores = self.get_mask_scores(reg_iou, det_bboxes, det_labels)
                 segm_result = self.get_seg_masks(
                     reg_pred_instance, det_bboxes[:, :4], det_labels, img_meta[0]['ori_shape'],
-                    img_meta[0]['scale_factor'], rescale=True)
+                    img_meta[0]['scale_factor'], rescale=rescale)
             else:
                 det_bboxes, det_labels = multiclass_nms1(det_bboxes[:, :4], det_bboxes[:, 4], det_labels + 1,
                                                          0.0, dict(type='soft_nms', iou_thr=0.5), 100)