Commit 65c3ebca authored by Kai Chen's avatar Kai Chen
Browse files

add min_pos_iou config field for train_cfg.rcnn

parent 323f430e
...@@ -57,7 +57,7 @@ class SingleLevelRoI(nn.Module): ...@@ -57,7 +57,7 @@ class SingleLevelRoI(nn.Module):
proposals = proposals[:, :4] proposals = proposals[:, :4]
assigned_gt_inds, assigned_labels, argmax_overlaps, max_overlaps = \ assigned_gt_inds, assigned_labels, argmax_overlaps, max_overlaps = \
bbox_assign(proposals, gt_bboxes, gt_crowds, gt_labels, 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: if cfg.add_gt_as_proposals:
proposals = torch.cat([gt_bboxes, proposals], dim=0) proposals = torch.cat([gt_bboxes, proposals], dim=0)
......
...@@ -65,6 +65,7 @@ train_cfg = dict( ...@@ -65,6 +65,7 @@ train_cfg = dict(
pos_balance_sampling=False, pos_balance_sampling=False,
neg_pos_ub=512, neg_pos_ub=512,
neg_balance_thr=0, neg_balance_thr=0,
min_pos_iou=1.1,
pos_weight=-1, pos_weight=-1,
debug=False)) debug=False))
test_cfg = dict( test_cfg = dict(
......
...@@ -77,6 +77,7 @@ train_cfg = dict( ...@@ -77,6 +77,7 @@ train_cfg = dict(
pos_balance_sampling=False, pos_balance_sampling=False,
neg_pos_ub=512, neg_pos_ub=512,
neg_balance_thr=0, neg_balance_thr=0,
min_pos_iou=1.1,
pos_weight=-1, pos_weight=-1,
debug=False)) debug=False))
test_cfg = dict( test_cfg = dict(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment