Upload 4 files
Browse files
codriving/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
codriving/perception/config.yaml
ADDED
@@ -0,0 +1,218 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
comm_range: 200
|
2 |
+
data_augment:
|
3 |
+
- ALONG_AXIS_LIST:
|
4 |
+
- x
|
5 |
+
NAME: random_world_flip
|
6 |
+
- NAME: random_world_rotation
|
7 |
+
WORLD_ROT_ANGLE:
|
8 |
+
- -0.78539816
|
9 |
+
- 0.78539816
|
10 |
+
- NAME: random_world_scaling
|
11 |
+
WORLD_SCALE_RANGE:
|
12 |
+
- 0.95
|
13 |
+
- 1.05
|
14 |
+
fusion:
|
15 |
+
args:
|
16 |
+
clip_pc: false
|
17 |
+
proj_first: false
|
18 |
+
core_method: intermediatemulticlass
|
19 |
+
dataset: v2xverse
|
20 |
+
input_source:
|
21 |
+
- lidar
|
22 |
+
label_type: lidar
|
23 |
+
loss:
|
24 |
+
args:
|
25 |
+
cls_weight: 20.0
|
26 |
+
code_weights:
|
27 |
+
- 1.0
|
28 |
+
- 1.0
|
29 |
+
- 1.0
|
30 |
+
- 1.0
|
31 |
+
- 1.0
|
32 |
+
- 1.0
|
33 |
+
- 20.0
|
34 |
+
- 20.0
|
35 |
+
loc_weight: 1.0
|
36 |
+
target_assigner_config:
|
37 |
+
box_coder: ResidualCoder
|
38 |
+
cav_lidar_range: &id004
|
39 |
+
- -36
|
40 |
+
- -12
|
41 |
+
- -22
|
42 |
+
- 36
|
43 |
+
- 12
|
44 |
+
- 14
|
45 |
+
gaussian_overlap: 0.1
|
46 |
+
max_objs: 40
|
47 |
+
min_radius: 2
|
48 |
+
out_size_factor: 2
|
49 |
+
voxel_size: &id001
|
50 |
+
- 0.125
|
51 |
+
- 0.125
|
52 |
+
- 36
|
53 |
+
core_method: center_point_loss_multiclass
|
54 |
+
lr_scheduler:
|
55 |
+
core_method: multistep
|
56 |
+
gamma: 0.1
|
57 |
+
step_size:
|
58 |
+
- 8
|
59 |
+
- 15
|
60 |
+
model:
|
61 |
+
args:
|
62 |
+
anchor_number: 3
|
63 |
+
base_bev_backbone:
|
64 |
+
compression: 0
|
65 |
+
layer_nums: &id002
|
66 |
+
- 3
|
67 |
+
- 4
|
68 |
+
- 5
|
69 |
+
layer_strides:
|
70 |
+
- 2
|
71 |
+
- 2
|
72 |
+
- 2
|
73 |
+
num_filters: &id003
|
74 |
+
- 64
|
75 |
+
- 128
|
76 |
+
- 256
|
77 |
+
num_upsample_filter:
|
78 |
+
- 128
|
79 |
+
- 128
|
80 |
+
- 128
|
81 |
+
resnet: true
|
82 |
+
upsample_strides:
|
83 |
+
- 1
|
84 |
+
- 2
|
85 |
+
- 4
|
86 |
+
voxel_size: *id001
|
87 |
+
fusion_args:
|
88 |
+
agg_operator:
|
89 |
+
feature_dim: 256
|
90 |
+
mode: ATTEN
|
91 |
+
communication:
|
92 |
+
thre: 0
|
93 |
+
cav_lidar_range: *id004
|
94 |
+
driving_request: False
|
95 |
+
gaussian_smooth:
|
96 |
+
k_size: 160
|
97 |
+
c_sigma: 5
|
98 |
+
downsample_rate: 1
|
99 |
+
dropout_rate: 0
|
100 |
+
in_channels: 256
|
101 |
+
layer_nums: *id002
|
102 |
+
multi_scale: true
|
103 |
+
n_head: 8
|
104 |
+
num_filters: *id003
|
105 |
+
only_attention: true
|
106 |
+
voxel_size: *id001
|
107 |
+
lidar_range: *id004
|
108 |
+
max_cav: 5
|
109 |
+
multi_class: true
|
110 |
+
out_size_factor: 2
|
111 |
+
pillar_vfe:
|
112 |
+
num_filters:
|
113 |
+
- 64
|
114 |
+
use_absolute_xyz: true
|
115 |
+
use_norm: true
|
116 |
+
with_distance: false
|
117 |
+
point_pillar_scatter:
|
118 |
+
grid_size: !!python/object/apply:numpy.core.multiarray._reconstruct
|
119 |
+
args:
|
120 |
+
- !!python/name:numpy.ndarray ''
|
121 |
+
- !!python/tuple
|
122 |
+
- 0
|
123 |
+
- !!binary |
|
124 |
+
Yg==
|
125 |
+
state: !!python/tuple
|
126 |
+
- 1
|
127 |
+
- !!python/tuple
|
128 |
+
- 3
|
129 |
+
- !!python/object/apply:numpy.dtype
|
130 |
+
args:
|
131 |
+
- i8
|
132 |
+
- false
|
133 |
+
- true
|
134 |
+
state: !!python/tuple
|
135 |
+
- 3
|
136 |
+
- <
|
137 |
+
- null
|
138 |
+
- null
|
139 |
+
- null
|
140 |
+
- -1
|
141 |
+
- -1
|
142 |
+
- 0
|
143 |
+
- false
|
144 |
+
- !!binary |
|
145 |
+
QAIAAAAAAADAAAAAAAAAAAEAAAAAAAAA
|
146 |
+
num_features: 64
|
147 |
+
shrink_header:
|
148 |
+
dim:
|
149 |
+
- 128
|
150 |
+
input_dim: 384
|
151 |
+
kernal_size:
|
152 |
+
- 3
|
153 |
+
padding:
|
154 |
+
- 1
|
155 |
+
stride:
|
156 |
+
- 1
|
157 |
+
supervise_single: true
|
158 |
+
voxel_size: *id001
|
159 |
+
core_method: center_point_codriving
|
160 |
+
name: v2xverse_codriving_ms_atten_multiclass
|
161 |
+
noise_setting: !!python/object/apply:collections.OrderedDict
|
162 |
+
- - - add_noise
|
163 |
+
- false
|
164 |
+
optimizer:
|
165 |
+
args:
|
166 |
+
eps: 1.0e-10
|
167 |
+
weight_decay: 0.0001
|
168 |
+
core_method: Adam
|
169 |
+
lr: 0.002
|
170 |
+
postprocess:
|
171 |
+
anchor_args:
|
172 |
+
D: 1
|
173 |
+
H: 192
|
174 |
+
W: 576
|
175 |
+
cav_lidar_range: *id004
|
176 |
+
feature_stride: 2
|
177 |
+
h: 1.56
|
178 |
+
l: 3.9
|
179 |
+
num: 1
|
180 |
+
r: &id005
|
181 |
+
- 0
|
182 |
+
vd: 36
|
183 |
+
vh: 0.125
|
184 |
+
vw: 0.125
|
185 |
+
w: 1.6
|
186 |
+
core_method: VoxelPostprocessor
|
187 |
+
dir_args:
|
188 |
+
anchor_yaw: *id005
|
189 |
+
dir_offset: 0.7853
|
190 |
+
num_bins: 1
|
191 |
+
gt_range: *id004
|
192 |
+
max_num: 100
|
193 |
+
nms_thresh: 0 # 0.15
|
194 |
+
order: hwl
|
195 |
+
target_args:
|
196 |
+
neg_threshold: 0.45
|
197 |
+
pos_threshold: 0.6
|
198 |
+
score_threshold: 0.2
|
199 |
+
preprocess:
|
200 |
+
args:
|
201 |
+
max_points_per_voxel: 32
|
202 |
+
max_voxel_test: 70000
|
203 |
+
max_voxel_train: 32000
|
204 |
+
voxel_size: *id001
|
205 |
+
cav_lidar_range: *id004
|
206 |
+
core_method: SpVoxelPreprocessor
|
207 |
+
root_dir: ./dataset/dataset_v2xverse/
|
208 |
+
test_dir: ./dataset/dataset_v2xverse/
|
209 |
+
train_params:
|
210 |
+
batch_size: 2 # 4
|
211 |
+
epoches: 40
|
212 |
+
eval_freq: 1
|
213 |
+
max_cav: 5
|
214 |
+
save_freq: 1
|
215 |
+
validate_dir: ./dataset/dataset_v2xverse/
|
216 |
+
yaml_parser: load_point_pillar_params
|
217 |
+
|
218 |
+
index_file: dataset_index
|
codriving/perception/net_epoch_bestval_at16.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f53047434e7336e09857f86ef9a92b55a194f60b0e284fe8f295bb89f87241ee
|
3 |
+
size 32817341
|
codriving/planner/codrving_planner.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:18642c160f9c60e2ea650a365df924c2b5d4f34b6c42b6844b81db51676ab038
|
3 |
+
size 20002291
|