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):
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)
......
......@@ -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(
......
......@@ -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(
......
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