From 627ecef014609f6148eaa8c689f0c96efeaae700 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bar=C4=B1=C5=9F=20Zeren?= <76053179+StepTurtle@users.noreply.github.com> Date: Mon, 27 Jan 2025 01:10:25 +0300 Subject: [PATCH] feat(ansible/artifacts): add rtmdet onnx model (#5605) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Barış Zeren Co-authored-by: Mete Fatih Cırıt --- ansible/roles/artifacts/tasks/main.yaml | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/ansible/roles/artifacts/tasks/main.yaml b/ansible/roles/artifacts/tasks/main.yaml index c4f7ede0089..f0423e71527 100644 --- a/ansible/roles/artifacts/tasks/main.yaml +++ b/ansible/roles/artifacts/tasks/main.yaml @@ -269,6 +269,28 @@ 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/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/tensorrt_rtmdet_onnx_models.tar.gz" + dest: "{{ data_dir }}/tensorrt_rtmdet/" + extra_opts: + - --strip-components=1 # Removes the top-level folder during extraction + # traffic_light_classifier - name: Create traffic_light_classifier directory inside {{ data_dir }} ansible.builtin.file: