From d579e8fb051178cd8424acb2dd874dcbdfc46baf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bar=C4=B1=C5=9F=20Zeren?= <baris@leodrive.ai>
Date: Mon, 23 Dec 2024 20:47:03 +0300
Subject: [PATCH] feat(ansible/artifacts): add rtmdet onnx model
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Barış Zeren <baris@leodrive.ai>
---
 ansible/roles/artifacts/tasks/main.yaml | 35 +++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/ansible/roles/artifacts/tasks/main.yaml b/ansible/roles/artifacts/tasks/main.yaml
index c4f7ede008..56fee326fd 100644
--- a/ansible/roles/artifacts/tasks/main.yaml
+++ b/ansible/roles/artifacts/tasks/main.yaml
@@ -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
+
+- name: Move extracted files to the final destination
+  ansible.builtin.copy:
+    src: "{{ data_dir }}/tensorrt_rtmdet/tensorrt_rtmdet_onnx_models/"
+    dest: "{{ data_dir }}/tensorrt_rtmdet/"
+    remote_src: yes
+
+- 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: