From 50c6ae692b9b27dc25bf209a1ddb06981bb7ae69 Mon Sep 17 00:00:00 2001
From: Jiangmiao Pang <pangjiangmiao@gmail.com>
Date: Tue, 4 Jun 2019 13:01:08 +0800
Subject: [PATCH] Fix bugs in the configs of Libra R-CNN (#738)
* Fix bugs in the configs of Libra R-CNN
* return configs to 2000 props
* return configs to 1000 props
---
configs/libra_rcnn/libra_fast_rcnn_r50_fpn_1x.py | 4 ++--
configs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x.py | 4 ++--
configs/libra_rcnn/libra_faster_rcnn_r50_fpn_1x.py | 4 ++--
configs/libra_rcnn/libra_faster_rcnn_x101_64x4d_fpn_1x.py | 4 ++--
4 files changed, 8 insertions(+), 8 deletions(-)
diff --git a/configs/libra_rcnn/libra_fast_rcnn_r50_fpn_1x.py b/configs/libra_rcnn/libra_fast_rcnn_r50_fpn_1x.py
index 30189a4..68ff1be 100644
--- a/configs/libra_rcnn/libra_fast_rcnn_r50_fpn_1x.py
+++ b/configs/libra_rcnn/libra_fast_rcnn_r50_fpn_1x.py
@@ -62,8 +62,8 @@ train_cfg = dict(
pos_sampler=dict(type='InstanceBalancedPosSampler'),
neg_sampler=dict(
type='IoUBalancedNegSampler',
- floor=-1,
- floor_thr=0,
+ floor_thr=-1,
+ floor_fraction=0,
num_bins=3)),
pos_weight=-1,
debug=False))
diff --git a/configs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x.py b/configs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x.py
index 3b7b2fb..7a31476 100644
--- a/configs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x.py
+++ b/configs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x.py
@@ -97,8 +97,8 @@ train_cfg = dict(
pos_sampler=dict(type='InstanceBalancedPosSampler'),
neg_sampler=dict(
type='IoUBalancedNegSampler',
- floor=-1,
- floor_thr=0,
+ floor_thr=-1,
+ floor_fraction=0,
num_bins=3)),
pos_weight=-1,
debug=False))
diff --git a/configs/libra_rcnn/libra_faster_rcnn_r50_fpn_1x.py b/configs/libra_rcnn/libra_faster_rcnn_r50_fpn_1x.py
index cec4944..d596903 100644
--- a/configs/libra_rcnn/libra_faster_rcnn_r50_fpn_1x.py
+++ b/configs/libra_rcnn/libra_faster_rcnn_r50_fpn_1x.py
@@ -97,8 +97,8 @@ train_cfg = dict(
pos_sampler=dict(type='InstanceBalancedPosSampler'),
neg_sampler=dict(
type='IoUBalancedNegSampler',
- floor=-1,
- floor_thr=0,
+ floor_thr=-1,
+ floor_fraction=0,
num_bins=3)),
pos_weight=-1,
debug=False))
diff --git a/configs/libra_rcnn/libra_faster_rcnn_x101_64x4d_fpn_1x.py b/configs/libra_rcnn/libra_faster_rcnn_x101_64x4d_fpn_1x.py
index 3b4a8e8..26172a4 100644
--- a/configs/libra_rcnn/libra_faster_rcnn_x101_64x4d_fpn_1x.py
+++ b/configs/libra_rcnn/libra_faster_rcnn_x101_64x4d_fpn_1x.py
@@ -99,8 +99,8 @@ train_cfg = dict(
pos_sampler=dict(type='InstanceBalancedPosSampler'),
neg_sampler=dict(
type='IoUBalancedNegSampler',
- floor=-1,
- floor_thr=0,
+ floor_thr=-1,
+ floor_fraction=0,
num_bins=3)),
pos_weight=-1,
debug=False))
--
GitLab