Skip to content

Commit

Permalink
feat(ansible/artifacts): add rtmdet onnx model
Browse files Browse the repository at this point in the history
Signed-off-by: Barış Zeren <[email protected]>
  • Loading branch information
StepTurtle committed Dec 23, 2024
1 parent c490050 commit d579e8f
Showing 1 changed file with 35 additions and 0 deletions.
35 changes: 35 additions & 0 deletions ansible/roles/artifacts/tasks/main.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -269,6 +269,41 @@
mode: "644"
checksum: sha256:3d93ca05f31b63424d7d7246a01a2365953705a0ed3323ba5b6fddd744a4bfea

# tensorrt_rtmdet
- name: Create tensorrt_rtmdet directory inside {{ data_dir }}
ansible.builtin.file:
path: "{{ data_dir }}/tensorrt_rtmdet"
mode: "755"
state: directory

- name: Download tensorrt_rtmdet_onnx_models.tar.gz
become: true
ansible.builtin.get_url:
url: https://autoware-files.s3.us-west-2.amazonaws.com/models/tensorrt_rtmdet_onnx_models.tar.gz
dest: "{{ data_dir }}/tensorrt_rtmdet_onnx_models.tar.gz"
mode: "644"
checksum: sha256:eaf6fe9caf1b4a0211dc3b0f4068b1fd6c1d35fe1582cc2aed95b8ed9468c598

- name: Extract tensorrt_rtmdet_onnx_models.tar.gz
ansible.builtin.unarchive:
src: "{{ data_dir }}/tensorrt_rtmdet_onnx_models.tar.gz"
dest: "{{ data_dir }}/tensorrt_rtmdet/"
remote_src: yes

Check failure on line 291 in ansible/roles/artifacts/tasks/main.yaml

View workflow job for this annotation

GitHub Actions / pre-commit-ansible

yaml[truthy]

Truthy value should be one of [false, true]

- name: Move extracted files to the final destination

Check failure on line 293 in ansible/roles/artifacts/tasks/main.yaml

View workflow job for this annotation

GitHub Actions / pre-commit-ansible

risky-file-permissions

File permissions unset or incorrect.
ansible.builtin.copy:
src: "{{ data_dir }}/tensorrt_rtmdet/tensorrt_rtmdet_onnx_models/"
dest: "{{ data_dir }}/tensorrt_rtmdet/"
remote_src: yes

Check failure on line 297 in ansible/roles/artifacts/tasks/main.yaml

View workflow job for this annotation

GitHub Actions / pre-commit-ansible

yaml[truthy]

Truthy value should be one of [false, true]

- name: Remove temporary files and directories
ansible.builtin.file:
path: "{{ item }}"
state: absent
with_items:
- "{{ data_dir }}/tensorrt_rtmdet_onnx_models.tar.gz"
- "{{ data_dir }}/tensorrt_rtmdet/tensorrt_rtmdet_onnx_models/"

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

0 comments on commit d579e8f

Please sign in to comment.