Skip to content

GitLab

Commit c136b5f4 authored by Kai Chen's avatar Kai Chen
Browse files

update training settings

parent 300f7157
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
tools/configs/r50_fpn_frcnn_1x.py
View file @ c136b5f4
...@@ -46,11 +46,12 @@ train_cfg = dict( ...@@ -46,11 +46,12 @@ train_cfg = dict(
pos_balance_sampling=False, pos_balance_sampling=False,
neg_pos_ub=256, neg_pos_ub=256,
allowed_border=0, allowed_border=0,
crowd_thr=1.1,
anchor_batch_size=256, anchor_batch_size=256,
pos_iou_thr=0.7, pos_iou_thr=0.7,
neg_iou_thr=0.3, neg_iou_thr=0.3,
neg_balance_thr=0, neg_balance_thr=0,
min_pos_iou=1e-3, min_pos_iou=0.3,
pos_weight=-1, pos_weight=-1,
smoothl1_beta=1 / 9.0, smoothl1_beta=1 / 9.0,
debug=False), debug=False),
......
tools/configs/r50_fpn_maskrcnn_1x.py
View file @ c136b5f4
...@@ -57,11 +57,12 @@ train_cfg = dict( ...@@ -57,11 +57,12 @@ train_cfg = dict(
pos_balance_sampling=False, pos_balance_sampling=False,
neg_pos_ub=256, neg_pos_ub=256,
allowed_border=0, allowed_border=0,
crowd_thr=1.1,
anchor_batch_size=256, anchor_batch_size=256,
pos_iou_thr=0.7, pos_iou_thr=0.7,
neg_iou_thr=0.3, neg_iou_thr=0.3,
neg_balance_thr=0, neg_balance_thr=0,
min_pos_iou=1e-3, min_pos_iou=0.3,
pos_weight=-1, pos_weight=-1,
smoothl1_beta=1 / 9.0, smoothl1_beta=1 / 9.0,
debug=False), debug=False),
......
tools/configs/r50_fpn_rpn_1x.py
View file @ c136b5f4
...@@ -31,11 +31,12 @@ train_cfg = dict( ...@@ -31,11 +31,12 @@ train_cfg = dict(
pos_balance_sampling=False, pos_balance_sampling=False,
neg_pos_ub=256, neg_pos_ub=256,
allowed_border=0, allowed_border=0,
crowd_thr=1.1,
anchor_batch_size=256, anchor_batch_size=256,
pos_iou_thr=0.7, pos_iou_thr=0.7,
neg_iou_thr=0.3, neg_iou_thr=0.3,
neg_balance_thr=0, neg_balance_thr=0,
min_pos_iou=1e-3, min_pos_iou=0.3,
pos_weight=-1, pos_weight=-1,
smoothl1_beta=1 / 9.0, smoothl1_beta=1 / 9.0,
debug=False)) debug=False))
......
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