Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
RODNet
Commits
aed6e05d
"vscode:/vscode.git/clone" did not exist on "fa2aa0db0afe09630c58b26ed5e29ea0b3dc936b"
Commit
aed6e05d
authored
Jan 02, 2021
by
Yizhou Wang
Browse files
Create config_rodnet_hg1_win16.py
parent
d73cfc99
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
76 additions
and
0 deletions
+76
-0
configs/config_rodnet_hg1_win16.py
configs/config_rodnet_hg1_win16.py
+76
-0
No files found.
configs/config_rodnet_hg1_win16.py
0 → 100644
View file @
aed6e05d
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
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment