fix: add download url of yolox with semantic segmentation

Signed-off-by: badai-nguyen <[email protected]>
autowarefoundation · Nov 21, 2023 · 6715a72 · 6715a72
1 parent c9c4546
commit 6715a72
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 1 deletion.
diff --git a/ansible/roles/artifacts/README.md b/ansible/roles/artifacts/README.md
@@ -66,7 +66,10 @@ $ wget -P ~/autoware_data/tensorrt_yolox/ \
        https://awf.ml.dev.web.auto/perception/models/yolox-sPlus-opt.EntropyV2-calibration.table \
        https://awf.ml.dev.web.auto/perception/models/object_detection_yolox_s/v1/yolox-sPlus-T4-960x960-pseudo-finetune.onnx \
        https://awf.ml.dev.web.auto/perception/models/object_detection_yolox_s/v1/yolox-sPlus-T4-960x960-pseudo-finetune.EntropyV2-calibration.table \
-       https://awf.ml.dev.web.auto/perception/models/label.txt
+       https://awf.ml.dev.web.auto/perception/models/label.txt \
+       https://awf.ml.dev.web.auto/perception/models/semantic_segmentation_yolox_s/v1/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.onnx \
+       https://awf.ml.dev.web.auto/perception/models/semantic_segmentation_yolox_s/v1/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.EntropyV2-calibration.table \
+       https://awf.ml.dev.web.auto/perception/models/semantic_segmentation_yolox_s/v1/semseg_color_map.csv
 
 
 # traffic_light_classifier

diff --git a/ansible/roles/artifacts/SHA256SUMS b/ansible/roles/artifacts/SHA256SUMS
@@ -24,6 +24,9 @@ f5054e8a890c3be86dc1b4b89a5a36fb2279d4f6110b0159e793be062641bf65  ./tensorrt_yol
 b9e9d7da33342262ccaea4469b4d02b8abb32b6d7bf737f9e0883fece1b8f580  ./tensorrt_yolox/yolox-sPlus-opt.EntropyV2-calibration.table
 36b0832177b01e6b278e00c7369f1de71e616c36261cbae50f0753d41289da01  ./tensorrt_yolox/yolox-sPlus-opt.onnx
 471a665f4243e654dff62578394e508db22ee29fe65d9e389dfc3b0f2dee1255  ./tensorrt_yolox/yolox-tiny.onnx
+4973da72e8c6420fca86ab8d61926df43776ffca3d3e3e39dc2a0297b3f83801  ./tensorrt_yolox/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.onnx
+fcd1a3535cad763af792b281b8d753a5bc0f18298af34f9b455fc56294228034  ./tensorrt_yolox/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.EntropyV2-calibration.table
+305224094e71feb74412a863e21ce4f870365d514284b691cba163eca22cf580  ./tensorrt_yolox/semseg_color_map.csv
 1a5a49eeec5593963eab8d70f48b8a01bfb07e753e9688eb1510ad26e803579d  ./traffic_light_classifier/lamp_labels.txt
 55ebb0d117a5e8943f8d1c6769f1d856b533079d4d871d8e923255cc992ad48a  ./traffic_light_classifier/traffic_light_classifier_efficientNet_b1_batch_1.onnx
 684e29843e3128eadb774018730644b3ab9b0a06dc4cdaeed579c2f3fa5d5265  ./traffic_light_classifier/traffic_light_classifier_efficientNet_b1_batch_4.onnx

diff --git a/ansible/roles/artifacts/tasks/main.yaml b/ansible/roles/artifacts/tasks/main.yaml
@@ -237,6 +237,30 @@
     mode: "644"
     checksum: sha256:3540a365bfd6d8afb1b5d8df4ec47f82cb984760d3270c9b41dbbb3422d09a0c
 
+- name: Download tensorrt_yolox/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.onnx
+  become: true
+  ansible.builtin.get_url:
+    url: https://awf.ml.dev.web.auto/perception/models/semantic_segmentation_yolox_s/v1/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.onnx
+    dest: "{{ data_dir }}/tensorrt_yolox/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.onnx"
+    mode: "644"
+    checksum: sha256:4973da72e8c6420fca86ab8d61926df43776ffca3d3e3e39dc2a0297b3f83801
+
+- name: Download tensorrt_yolox/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.EntropyV2-calibration.table
+  become: true
+  ansible.builtin.get_url:
+    url: https://awf.ml.dev.web.auto/perception/models/semantic_segmentation_yolox_s/v1/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.EntropyV2-calibration.table
+    dest: "{{ data_dir }}/tensorrt_yolox/yoloXP-sPlus-T4-960x960-pseudo-finetune-semseg.EntropyV2-calibration.table"
+    mode: "644"
+    checksum: sha256:fcd1a3535cad763af792b281b8d753a5bc0f18298af34f9b455fc56294228034
+
+- name: Download tensorrt_yolox/semseg_color_map.csv
+  become: true
+  ansible.builtin.get_url:
+    url: https://awf.ml.dev.web.auto/perception/models/semantic_segmentation_yolox_s/v1/semseg_color_map.csv
+    dest: "{{ data_dir }}/tensorrt_yolox/semseg_color_map.csv"
+    mode: "644"
+    checksum: sha256:305224094e71feb74412a863e21ce4f870365d514284b691cba163eca22cf580
+
 # traffic_light_classifier
 - name: Create traffic_light_classifier directory inside {{ data_dir }}
   ansible.builtin.file: