Skip to content

Commit

Permalink
feat: add new traffic light detector artifacts download (#5764)
Browse files Browse the repository at this point in the history
Signed-off-by: badai-nguyen <[email protected]>
Signed-off-by: MasatoSaeki <[email protected]>
  • Loading branch information
MasatoSaeki authored Feb 18, 2025
1 parent 4374782 commit cad73fb
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions ansible/roles/artifacts/tasks/main.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -245,6 +245,7 @@
mode: "644"
checksum: sha256:3540a365bfd6d8afb1b5d8df4ec47f82cb984760d3270c9b41dbbb3422d09a0c

# cspell: ignore semseg
- name: Download tensorrt_yolox/yolox-sPlus-opt-pseudoV2-T4-960x960-T4-seg16cls.onnx
become: true
ansible.builtin.get_url:
Expand Down Expand Up @@ -291,6 +292,31 @@
extra_opts:
- --strip-components=1 # Removes the top-level folder during extraction

# tensorrt_yolox for whole_image_traffic_light_detector
- name: Download tensorrt_yolox/yolox_s_car_ped_tl_detector_960_960_batch_1.onnx
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/tl_detector_yolox_s/v1/yolox_s_car_ped_tl_detector_960_960_batch_1.onnx
dest: "{{ data_dir }}/tensorrt_yolox/yolox_s_car_ped_tl_detector_960_960_batch_1.onnx"
mode: "644"
checksum: sha256:0b8478553f2f0374a1236e65f669f11335b1fea7756ca7bcdfcf9646657ed594

- name: Download tensorrt_yolox/yolox_s_car_ped_tl_detector_960_960_batch_1.EntropyV2-calibration.table
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/tl_detector_yolox_s/v1/yolox_s_car_ped_tl_detector_960_960_batch_1.EntropyV2-calibration.table
dest: "{{ data_dir }}/tensorrt_yolox/yolox_s_car_ped_tl_detector_960_960_batch_1.EntropyV2-calibration.table"
mode: "644"
checksum: sha256:690cff409519aca26f4ec0fc56ed35d9aa23ee1c18d6205c43469ae06e2f4e02

- name: Download tensorrt/car_ped_tl_detector_labels.txt
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/tl_detector_yolox_s/v1/car_ped_tl_detector_labels.txt
dest: "{{ data_dir }}/tensorrt_yolox/car_ped_tl_detector_labels.txt"
mode: "644"
checksum: sha256:a2a91f5fe9c2e68e3e3647a272bb9bb25cd07631a1990a3fb15efddce7691131

# traffic_light_classifier
- name: Create traffic_light_classifier directory inside {{ data_dir }}
ansible.builtin.file:
Expand Down

0 comments on commit cad73fb

Please sign in to comment.