Skip to content

Commit 5ae1125

Browse files
authored
feat(ansible): add VAD model download configuration (#6557)
* feat(ansible): add VAD model download configuration Add Ansible tasks to download VAD CARLA Tiny model files: - vad-carla-tiny_backbone.onnx - vad-carla-tiny_head.onnx - vad-carla-tiny_head_no_prev.onnx - vad-carla-tiny.param.json Models are downloaded to autoware_data/vad/ directory with SHA256 checksums for verification. Signed-off-by: Max-Bin <vborisw@gmail.com> * feat(ansible): add version directory for VAD model downloads Create v0.1 subdirectory for VAD models to support version management. This enables Ansible to properly handle model updates without overwriting existing model versions. Signed-off-by: Max-Bin <vborisw@gmail.com> * fix(ansible): update checksum for VAD model param file Update checksum for vad-carla-tiny.param.json to match the current file available on the server. The file content was updated upstream, causing the previous checksum validation to fail. Signed-off-by: Max-Bin <vborisw@gmail.com> --------- Signed-off-by: Max-Bin <vborisw@gmail.com>
1 parent f55e68a commit 5ae1125

1 file changed

Lines changed: 45 additions & 0 deletions

File tree

ansible/roles/artifacts/tasks/main.yaml

Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -538,6 +538,51 @@
538538
mode: "644"
539539
checksum: sha256:bf41f12d85b31e363acf84f46d6a9e4e28da6c2ccce0e3fcf91ee4910f4b8453
540540

541+
# tensorrt_vad
542+
- name: Create tensorrt_vad directory inside {{ data_dir }}
543+
ansible.builtin.file:
544+
path: "{{ data_dir }}/vad"
545+
mode: "755"
546+
state: directory
547+
548+
- name: Create tensorrt_vad v0.1 directory inside {{ data_dir }}
549+
ansible.builtin.file:
550+
path: "{{ data_dir }}/vad/v0.1"
551+
mode: "755"
552+
state: directory
553+
554+
- name: Download tensorrt_vad/vad-carla-tiny_backbone.onnx
555+
become: true
556+
ansible.builtin.get_url:
557+
url: https://awf.ml.dev.web.auto/planning/models/tensorrt_vad/carla_tiny/v0.1/vad-carla-tiny_backbone.onnx
558+
dest: "{{ data_dir }}/vad/v0.1/vad-carla-tiny_backbone.onnx"
559+
mode: "644"
560+
checksum: sha256:04b925f2750fd1c4adf16b5aae9c149d0baa39185e99d141232ebe20bddba4da
561+
562+
- name: Download tensorrt_vad/vad-carla-tiny_head.onnx
563+
become: true
564+
ansible.builtin.get_url:
565+
url: https://awf.ml.dev.web.auto/planning/models/tensorrt_vad/carla_tiny/v0.1/vad-carla-tiny_head.onnx
566+
dest: "{{ data_dir }}/vad/v0.1/vad-carla-tiny_head.onnx"
567+
mode: "644"
568+
checksum: sha256:31f49a592a764ce82bbe6e26d0bfc99dc8a9613628884dc73dd5e67521ff3e9e
569+
570+
- name: Download tensorrt_vad/vad-carla-tiny_head_no_prev.onnx
571+
become: true
572+
ansible.builtin.get_url:
573+
url: https://awf.ml.dev.web.auto/planning/models/tensorrt_vad/carla_tiny/v0.1/vad-carla-tiny_head_no_prev.onnx
574+
dest: "{{ data_dir }}/vad/v0.1/vad-carla-tiny_head_no_prev.onnx"
575+
mode: "644"
576+
checksum: sha256:6a89d479e0717b1e526f1aa3a1137c631a9ef854e810d0328a035aae11818c2a
577+
578+
- name: Download tensorrt_vad/vad-carla-tiny.param.json
579+
become: true
580+
ansible.builtin.get_url:
581+
url: https://awf.ml.dev.web.auto/planning/models/tensorrt_vad/carla_tiny/v0.1/vad-carla-tiny.param.json
582+
dest: "{{ data_dir }}/vad/v0.1/vad-carla-tiny.param.json"
583+
mode: "644"
584+
checksum: sha256:03d3187fed3c70f761456afc2e93e18c1765113b1c1580ffa78ab42b10dbd179
585+
541586
# traffic_light_fine_detector
542587
- name: Create traffic_light_fine_detector directory inside {{ data_dir }}
543588
ansible.builtin.file:

0 commit comments

Comments
 (0)