-
Notifications
You must be signed in to change notification settings - Fork 6
/
hd_240m.py
54 lines (43 loc) · 816 Bytes
/
hd_240m.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# DATA
dataset='nuScenes'
dataroot = './dataset/nuScenes'
version= 'v1.0-trainval'
xbound = [-120.0, 120.0, 0.3] #240m*60m, bev_size:800*200
ybound = [-30.0, 30.0, 0.3]
zbound = [-10.0, 10.0, 20.0]
dbound = [4.0, 45.0, 1.0]
image_size = [128, 352]
thickness = 5
# EXP
logdir = './Work_dir/hd_240'
sd_map_path='./data_osm/osm'
# TRAIN
model = 'pmapnet_hd'
nepochs = 30
batch_size = 8
nworkers = 20
gpus = [0, 1, 2, 3]
# OPT
lr = 5e-4
weight_decay = 1e-7
max_grad_norm = 5.0
pos_weight = 2.13
steplr = 10
# CHECK_POINTS
modelf = None
# LOSS
scale_seg = 1.0
scale_var = 0.1
scale_dist = 0.1
scale_direction = 0.1
direction_pred = True
instance_seg = True
embedding_dim = 16
delta_v = 0.5
delta_d = 3.0
angle_class = 36
# Mask config
mask_flag = False
mask_ratio = -1 # random ratio
patch_h = 20
patch_w = 20