"tests/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "71f34fc5a4d5611ba26bc1e848d5afd050ffdbcc"
Commit aed6e05d authored by Yizhou Wang's avatar Yizhou Wang
Browse files

Create config_rodnet_hg1_win16.py

parent d73cfc99
dataset_cfg = dict(
dataset_name='ROD2021',
base_root="/mnt/disk2/CRUW/ROD2021",
data_root="/mnt/disk2/CRUW/ROD2021/sequences",
anno_root="/mnt/disk2/CRUW/ROD2021/annotations",
anno_ext='.txt',
train=dict(
subdir='train',
# seqs=[], # can choose from the subdir folder
),
valid=dict(
subdir='valid',
seqs=[],
),
test=dict(
subdir='test',
# seqs=[], # can choose from the subdir folder
),
demo=dict(
subdir='demo',
seqs=[],
),
)
model_cfg = dict(
type='HG',
name='rodnet-hg1-win16-wobg',
max_dets=20,
peak_thres=0.4,
ols_thres=0.3,
stacked_num=1,
)
confmap_cfg = dict(
confmap_sigmas={
'pedestrian': 15,
'cyclist': 20,
'car': 30,
# 'van': 40,
# 'truck': 50,
},
confmap_sigmas_interval={
'pedestrian': [5, 15],
'cyclist': [8, 20],
'car': [10, 30],
# 'van': [15, 40],
# 'truck': [20, 50],
},
confmap_length={
'pedestrian': 1,
'cyclist': 2,
'car': 3,
# 'van': 4,
# 'truck': 5,
}
)
train_cfg = dict(
n_epoch=50,
batch_size=4,
lr=0.00001,
lr_step=5, # lr will decrease 10 times after lr_step epoches
win_size=16,
train_step=1,
train_stride=4,
log_step=100,
save_step=10000,
)
test_cfg = dict(
test_step=1,
test_stride=8,
rr_min=1.0, # min radar range
rr_max=20.0, # max radar range
ra_min=-60.0, # min radar angle
ra_max=60.0, # max radar angle
)
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