From 65c3ebca03aa052cbb506493afb216318875fad7 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Wed, 3 Oct 2018 17:31:17 +0800
Subject: [PATCH] add min_pos_iou config field for train_cfg.rcnn
---
mmdet/models/roi_extractors/single_level.py | 2 +-
tools/configs/r50_fpn_frcnn_1x.py | 1 +
tools/configs/r50_fpn_maskrcnn_1x.py | 1 +
3 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/mmdet/models/roi_extractors/single_level.py b/mmdet/models/roi_extractors/single_level.py
index fa247a5..b850d6a 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 83ba65b..e15cbdb 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 2fa7199..5ecdaf4 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(
--
GitLab