Skip to content

GitLab

Commit 81f1e0ac authored by Yizhou Wang's avatar Yizhou Wang
Browse files

Update config_rodnet_cdc_win16.py

parent 07ebf4f8
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 18 deletions
configs/config_rodnet_cdc_win16.py
View file @ 81f1e0ac
dataset_cfg = dict( dataset_cfg = dict(
base_root="/mnt/disk2/CRUW/CRUW_MINI", dataset_name='ROD2021',
data_root="/mnt/disk2/CRUW/CRUW_MINI/sequences", base_root="/mnt/disk1/CRUW/ROD2021",
anno_root="/mnt/disk2/CRUW/CRUW_MINI/annotations", data_root="/mnt/disk1/CRUW/ROD2021/sequences",
anno_root="/mnt/disk1/CRUW/ROD2021/annotations",
anno_ext='.txt',
train=dict( train=dict(
seqs=[ subdir='train',
'2019_04_09_BMS1000_PL_NORMAL', # seqs=[], # can choose from the subdir folder
'2019_04_09_CMS1002_PL_NORMAL',
'2019_04_09_PMS1000_PL_NORMAL',
'2019_04_09_PMS3001_PL_NORMAL',
'2019_05_29_MLMS006_CR_BLUR',
'2019_09_29_ONRD001_CS_NORMAL',
'2019_09_29_ONRD002_CS_NORMAL',
'2019_09_29_ONRD004_HW_NORMAL',
],
), ),
valid=dict( valid=dict(
subdir='valid',
seqs=[], seqs=[],
), ),
test=dict( test=dict(
seqs=[ subdir='test',
'2019_05_29_PBMS007_PL_BLUR', # seqs=[], # can choose from the subdir folder
'2019_10_13_ONRD048_CS_NIGHT'
],
), ),
demo=dict( demo=dict(
subdir='demo',
seqs=[], seqs=[],
), ),
) )
...@@ -67,7 +61,7 @@ train_cfg = dict( ...@@ -67,7 +61,7 @@ train_cfg = dict(
lr_step=5, # lr will decrease 10 times after lr_step epoches lr_step=5, # lr will decrease 10 times after lr_step epoches
win_size=16, win_size=16,
train_step=1, train_step=1,
train_stride=16, train_stride=4,
log_step=100, log_step=100,
save_step=1000, save_step=1000,
) )
......
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