diff --git a/mmdet/models/roi_extractors/single_level.py b/mmdet/models/roi_extractors/single_level.py index fa247a520be35abcd1e3b24135c4159e42ef4f95..b850d6ab680eaf0ef3a03890d75181d570d3b9f4 100644 --- a/mmdet/models/roi_extractors/single_level.py +++ b/mmdet/models/roi_extractors/single_level.py @@ -57,7 +57,7 @@ class SingleLevelRoI(nn.Module): proposals = proposals[:, :4] assigned_gt_inds, assigned_labels, argmax_overlaps, max_overlaps = \ bbox_assign(proposals, gt_bboxes, gt_crowds, gt_labels, - cfg.pos_iou_thr, cfg.neg_iou_thr, cfg.pos_iou_thr, cfg.crowd_thr) + cfg.pos_iou_thr, cfg.neg_iou_thr, cfg.min_pos_iou, cfg.crowd_thr) if cfg.add_gt_as_proposals: proposals = torch.cat([gt_bboxes, proposals], dim=0) diff --git a/tools/configs/r50_fpn_frcnn_1x.py b/tools/configs/r50_fpn_frcnn_1x.py index 83ba65b896c32d9292d22e6791bd6c9ed87820df..e15cbdbfec52795ceb9ec12ef30f86d261e8f608 100644 --- a/tools/configs/r50_fpn_frcnn_1x.py +++ b/tools/configs/r50_fpn_frcnn_1x.py @@ -65,6 +65,7 @@ train_cfg = dict( pos_balance_sampling=False, neg_pos_ub=512, neg_balance_thr=0, + min_pos_iou=1.1, pos_weight=-1, debug=False)) test_cfg = dict( diff --git a/tools/configs/r50_fpn_maskrcnn_1x.py b/tools/configs/r50_fpn_maskrcnn_1x.py index 2fa719980cfdf063caa7bbaf1b9d620648930b7d..5ecdaf44da91030efdd734c031499a6d7d7a66f5 100644 --- a/tools/configs/r50_fpn_maskrcnn_1x.py +++ b/tools/configs/r50_fpn_maskrcnn_1x.py @@ -77,6 +77,7 @@ train_cfg = dict( pos_balance_sampling=False, neg_pos_ub=512, neg_balance_thr=0, + min_pos_iou=1.1, pos_weight=-1, debug=False)) test_cfg = dict(