Skip to content
Snippets Groups Projects
Commit e032ebb7 authored by Jiangmiao Pang's avatar Jiangmiao Pang Committed by Kai Chen
Browse files

Update Libra RetinaNet config with the latest code (#1975)

* Fix IoU sampling

* Update config of libra rcnn with the latest code
parent 8c0ecd1e
No related branches found
No related tags found
No related merge requests found
...@@ -15,7 +15,6 @@ model = dict( ...@@ -15,7 +15,6 @@ model = dict(
in_channels=[256, 512, 1024, 2048], in_channels=[256, 512, 1024, 2048],
out_channels=256, out_channels=256,
start_level=1, start_level=1,
extra_convs_on_inputs=True,
add_extra_convs=True, add_extra_convs=True,
num_outs=5), num_outs=5),
dict( dict(
...@@ -57,9 +56,6 @@ train_cfg = dict( ...@@ -57,9 +56,6 @@ train_cfg = dict(
neg_iou_thr=0.4, neg_iou_thr=0.4,
min_pos_iou=0, min_pos_iou=0,
ignore_iof_thr=-1), ignore_iof_thr=-1),
smoothl1_beta=0.11,
gamma=2.0,
alpha=0.25,
allowed_border=-1, allowed_border=-1,
pos_weight=-1, pos_weight=-1,
debug=False) debug=False)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment