pshashid commited on
Commit
ee7e914
·
verified ·
1 Parent(s): d4bce28

Add/update vad-autoware

Browse files
vad-autoware/deploy_metadata.yaml ADDED
@@ -0,0 +1 @@
 
 
1
+ version: v0.1
vad-autoware/meta.json ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "vad-autoware",
3
+ "source": "AutowareFoundation/tensorrt_vad",
4
+ "method": "prebuilt",
5
+ "note": "CARLA-trained VAD, not the nuScenes original",
6
+ "revision": "1539e852f94d7a9afdc32b90cf65149fd0b429aa",
7
+ "validation": "prebuilt, not re-validated",
8
+ "opset_requested": 17,
9
+ "files": {
10
+ "vad-carla-tiny_backbone.onnx": {
11
+ "sha256": "04b925f2750fd1c4adf16b5aae9c149d0baa39185e99d141232ebe20bddba4da",
12
+ "bytes": 98422204,
13
+ "inputs": {
14
+ "img": [
15
+ 6,
16
+ 3,
17
+ 384,
18
+ 640
19
+ ]
20
+ },
21
+ "outputs": {
22
+ "out.0": [
23
+ 6,
24
+ 1,
25
+ 256,
26
+ 12,
27
+ 20
28
+ ]
29
+ },
30
+ "opset": 15
31
+ },
32
+ "vad-carla-tiny_head.onnx": {
33
+ "sha256": "31f49a592a764ce82bbe6e26d0bfc99dc8a9613628884dc73dd5e67521ff3e9e",
34
+ "bytes": 149881872,
35
+ "inputs": {
36
+ "mlvl_feats.0": [
37
+ 1,
38
+ 6,
39
+ 256,
40
+ 12,
41
+ 20
42
+ ],
43
+ "img_metas.0[shift]": [
44
+ 1,
45
+ 2
46
+ ],
47
+ "img_metas.0[lidar2img]": [
48
+ 6,
49
+ 4,
50
+ 4
51
+ ],
52
+ "img_metas.0[can_bus]": [
53
+ 1,
54
+ 18
55
+ ],
56
+ "prev_bev": [
57
+ 21200,
58
+ 1,
59
+ 256
60
+ ]
61
+ },
62
+ "outputs": {
63
+ "out.bev_embed": [
64
+ 21200,
65
+ 1,
66
+ 256
67
+ ],
68
+ "out.all_cls_scores": [
69
+ 3,
70
+ 1,
71
+ 300,
72
+ 9
73
+ ],
74
+ "out.all_bbox_preds": [
75
+ 3,
76
+ 1,
77
+ 300,
78
+ 10
79
+ ],
80
+ "out.all_traj_preds": [
81
+ 3,
82
+ 1,
83
+ 300,
84
+ 6,
85
+ 12
86
+ ],
87
+ "out.all_traj_cls_scores": [
88
+ 3,
89
+ 1,
90
+ 300,
91
+ 6
92
+ ],
93
+ "out.map_all_cls_scores": [
94
+ 3,
95
+ 1,
96
+ 100,
97
+ 6
98
+ ],
99
+ "out.map_all_bbox_preds": [
100
+ 3,
101
+ 1,
102
+ 100,
103
+ 4
104
+ ],
105
+ "out.map_all_pts_preds": [
106
+ 3,
107
+ 1,
108
+ 100,
109
+ 20,
110
+ 2
111
+ ],
112
+ "out.ego_fut_preds": [
113
+ 1,
114
+ 6,
115
+ 6,
116
+ 2
117
+ ]
118
+ },
119
+ "opset": 15
120
+ },
121
+ "vad-carla-tiny_head_no_prev.onnx": {
122
+ "sha256": "6a89d479e0717b1e526f1aa3a1137c631a9ef854e810d0328a035aae11818c2a",
123
+ "bytes": 149883556,
124
+ "inputs": {
125
+ "mlvl_feats.0": [
126
+ 1,
127
+ 6,
128
+ 256,
129
+ 12,
130
+ 20
131
+ ],
132
+ "img_metas.0[shift]": [
133
+ 1,
134
+ 2
135
+ ],
136
+ "img_metas.0[lidar2img]": [
137
+ 6,
138
+ 4,
139
+ 4
140
+ ],
141
+ "img_metas.0[can_bus]": [
142
+ 1,
143
+ 18
144
+ ]
145
+ },
146
+ "outputs": {
147
+ "out.bev_embed": [
148
+ 21200,
149
+ 1,
150
+ 256
151
+ ],
152
+ "out.all_cls_scores": [
153
+ 3,
154
+ 1,
155
+ 300,
156
+ 9
157
+ ],
158
+ "out.all_bbox_preds": [
159
+ 3,
160
+ 1,
161
+ 300,
162
+ 10
163
+ ],
164
+ "out.all_traj_preds": [
165
+ 3,
166
+ 1,
167
+ 300,
168
+ 6,
169
+ 12
170
+ ],
171
+ "out.all_traj_cls_scores": [
172
+ 3,
173
+ 1,
174
+ 300,
175
+ 6
176
+ ],
177
+ "out.map_all_cls_scores": [
178
+ 3,
179
+ 1,
180
+ 100,
181
+ 6
182
+ ],
183
+ "out.map_all_bbox_preds": [
184
+ 3,
185
+ 1,
186
+ 100,
187
+ 4
188
+ ],
189
+ "out.map_all_pts_preds": [
190
+ 3,
191
+ 1,
192
+ 100,
193
+ 20,
194
+ 2
195
+ ],
196
+ "out.ego_fut_preds": [
197
+ 1,
198
+ 6,
199
+ 6,
200
+ 2
201
+ ]
202
+ },
203
+ "opset": 15
204
+ }
205
+ },
206
+ "exported_at": "2026-10-19T18:45:49.219773+00:00"
207
+ }
vad-autoware/vad-carla-tiny.param.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "major_version": 0,
3
+ "minor_version": 1,
4
+ "model_name": "vad-carla-tiny",
5
+ "training_dataset": "Bench2Drive CARLA",
6
+ "release_date": "2025-11-04",
7
+ "description": "VAD model trained on Bench2Drive CARLA dataset with 6-camera surround view",
8
+
9
+ "image_normalization": {
10
+ "mean": [123.675, 116.28, 103.53],
11
+ "std": [58.395, 57.12, 57.375],
12
+ "format": "RGB",
13
+ "note": "ImageNet normalization statistics"
14
+ },
15
+
16
+ "input_specs": {
17
+ "target_image_width": 640,
18
+ "target_image_height": 384,
19
+ "num_cameras": 6,
20
+ "camera_order": ["FRONT", "BACK", "FRONT_LEFT", "BACK_LEFT", "FRONT_RIGHT", "BACK_RIGHT"]
21
+ },
22
+
23
+ "class_definitions": {
24
+ "map_classes": ["Broken", "Solid", "SolidSolid", "Center", "TrafficLight", "StopSign"],
25
+ "object_classes": ["car", "van", "truck", "bicycle", "traffic_sign", "traffic_cone", "traffic_light", "pedestrian", "others"]
26
+ },
27
+
28
+ "network_architecture": {
29
+ "bev": {
30
+ "height": 106,
31
+ "width": 200,
32
+ "feature_dim": 256,
33
+ "downsample_factor": 32
34
+ },
35
+ "transformer": {
36
+ "num_decoder_layers": 3
37
+ },
38
+ "prediction": {
39
+ "num_queries": 300,
40
+ "num_classes": 9,
41
+ "bbox_pred_dim": 10,
42
+ "trajectory_modes": 6,
43
+ "timesteps": 6,
44
+ "bbox_format": "[cx, cy, w, l, cz, h, sin, cos, vx, vy]"
45
+ },
46
+ "planning": {
47
+ "ego_commands": 6,
48
+ "timesteps": 6
49
+ },
50
+ "map": {
51
+ "num_queries": 100,
52
+ "num_classes": 6,
53
+ "points_per_polyline": 20
54
+ },
55
+ "vehicle_state": {
56
+ "can_bus_dim": 18
57
+ }
58
+ },
59
+
60
+ "model_files": {
61
+ "backbone": "vad-carla-tiny_backbone.onnx",
62
+ "head": "vad-carla-tiny_head.onnx",
63
+ "head_no_prev": "vad-carla-tiny_head_no_prev.onnx"
64
+ },
65
+
66
+ "compatibility": {
67
+ "autoware_tensorrt_vad_version": ">= 0.1.0",
68
+ "tensorrt_version": ">= 8.6",
69
+ "cuda_version": ">= 11.4"
70
+ }
71
+ }
vad-autoware/vad-carla-tiny_backbone.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04b925f2750fd1c4adf16b5aae9c149d0baa39185e99d141232ebe20bddba4da
3
+ size 98422204
vad-autoware/vad-carla-tiny_head.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31f49a592a764ce82bbe6e26d0bfc99dc8a9613628884dc73dd5e67521ff3e9e
3
+ size 149881872
vad-autoware/vad-carla-tiny_head_no_prev.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a89d479e0717b1e526f1aa3a1137c631a9ef854e810d0328a035aae11818c2a
3
+ size 149883556