Updates :

KBOUSTM · KBOUSTM · commit 9f6336ce691f · 2025-01-07T17:02:24.000+01:00
- Fix #1: Issue with the version parsing - Correct the YOLOv8n model paths in the yaml files Signed-off-by: khaoula boutiche <khaoula.boutiche@st.com>
diff --git a/common/stm32ai_local/stm_ai_tools.py b/common/stm32ai_local/stm_ai_tools.py
@@ -98,19 +98,10 @@ def _get_stm32ai_cli(root_path: Union[str, Path], host_os: str) -> List[Tuple[ST
             middlewares = cdt.parent / 'Middlewares'
         cmd_line = str(filepath) + ' --version'
         err, lines = run_shell_cmd(cmd_line, logger=logger)
-        pattern_8 = re.compile(r'STM.ai v(\d+\.\d+\.\d+)')
-        version_8 = pattern_8.findall(lines[0])
-        pattern_9 = re.compile(r'STM32 MCU module v(\d+\.\d+\.\d+)')
-        version_9 = pattern_9.findall(lines[1]) if len(lines) > 1 else []
-        pattern_10= re.compile(r'STM32CubeAI (\d+\.\d+\.\d+)')
-        version_10 = pattern_10.findall(lines[4]) if len(lines) > 4 else []
-        if version_8 != []:
-            version = version_8
-        elif version_9 != []:
-            version = version_9
-        elif version_10 != []:
-            version = version_10
-        else:
+        pattern = re.compile(r'STM32CubeAI (\d+\.\d+\.\d+)')
+        lines_conc = "".join(lines)
+        version = pattern.findall(lines_conc)
+        if version == []:
             raise ValueError('Unable to find the CubeAi version')
         ver = STMAiVersion(version[0], '{} pack'.format(_X_CUBE_AI_PACK_TAG))
         results.append((ver, cdt.parents[idx], cdt))
diff --git a/common/utils/cfg_utils.py b/common/utils/cfg_utils.py
@@ -14,6 +14,7 @@
 from pathlib import Path
 import re
 import numpy as np
+import requests
 
 aspect_ratio_dict = {"fit": "ASPECT_RATIO_FIT",
                      "crop": "ASPECT_RATIO_CROP",
@@ -23,8 +24,6 @@
 color_mode_n6_dict = {"rgb": "COLOR_RGB",
                       "bgr": "COLOR_BGR"}
                       
-                      
-import requests
 
 
 def download_file(url:str, local_path:str):
diff --git a/instance_segmentation/src/benchmarking/README.md b/instance_segmentation/src/benchmarking/README.md
@@ -19,7 +19,7 @@ In particular, `operation_mode` should be set to evaluation and the `benchmarkin
 general:
   project_name: coco_instance_seg
   model_type: yolo_v8_seg
-  model_path: path/to/stm32-hotspot/ultralytics/yolov8n-seg_full_integer_quant_256.tflite
+  model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/segmentation/yolov8n_256_quant_pc_uf_seg_coco-st.tflite
 
 operation_mode: benchmarking
 ```
diff --git a/instance_segmentation/src/config_file_examples/benchmarking_config.yaml b/instance_segmentation/src/config_file_examples/benchmarking_config.yaml
@@ -1,7 +1,7 @@
 general:
   project_name: coco_instance_seg
   model_type: yolo_v8_seg
-  model_path: path/to/stm32-hotspot/ultralytics/yolov8n-seg_full_integer_quant_256.tflite
+  model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/segmentation/yolov8n_256_quant_pc_uf_seg_coco-st.tflite
 
 operation_mode: benchmarking
 
diff --git a/instance_segmentation/src/config_file_examples/deployment_n6_config.yaml b/instance_segmentation/src/config_file_examples/deployment_n6_config.yaml
@@ -1,7 +1,7 @@
 general:
   project_name: coco_instance_seg
   model_type: yolo_v8_seg
-  model_path: path/to/stm32-hotspot/ultralytics/yolov8n_256_quant_pc_uf_seg_coco-st.tflite
+  model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/segmentation/yolov8n_256_quant_pc_uf_seg_coco-st.tflite
 
 operation_mode: deployment
 
diff --git a/instance_segmentation/src/config_file_examples/deployment_n6_yolov8n_seg_config.yaml b/instance_segmentation/src/config_file_examples/deployment_n6_yolov8n_seg_config.yaml
@@ -1,7 +1,7 @@
 general:
   model_type: yolo_v8_seg # yolo_v8_seg
   # path to a `.tflite` or `.onnx` file.
-  model_path: ../../../stm32ai-modelzoo/instance_segmentation/yolov8n_seg/yolov8n_256_quant_pc_uf_seg_coco-st.tflite
+  model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/segmentation/yolov8n_256_quant_pc_uf_seg_coco-st.tflite
 
 operation_mode: deployment
 
diff --git a/instance_segmentation/src/config_file_examples/prediction_config.yaml b/instance_segmentation/src/config_file_examples/prediction_config.yaml
@@ -1,7 +1,7 @@
 general:
   project_name: coco_instance_seg
   model_type: yolo_v8_seg
-  model_path: path/to/stm32-hotspot/ultralytics/yolov8n-seg_full_integer_quant_256.tflite
+  model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/segmentation/yolov8n_256_quant_pc_uf_seg_coco-st.tflite
   display_figures: true
 
 operation_mode: prediction
diff --git a/instance_segmentation/src/prediction/README.md b/instance_segmentation/src/prediction/README.md
@@ -24,7 +24,7 @@ The `general` section and its attributes are shown below.
 ```yaml
 general:
   project_name: coco_instance_seg          # Project name. Optional, defaults to "<unnamed>".
-  model_path: path/to/stm32-hotspot/ultralytics/yolov8n-seg_full_integer_quant_256.tflite
+  model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/segmentation/yolov8n_256_quant_pc_uf_seg_coco-st.tflite
   gpu_memory_limit: 16                     # Maximum amount of GPU memory in GBytes that TensorFlow may use (an integer).
 ```
 
diff --git a/object_detection/src/config_file_examples/deployment_n6_yolov8_config.yaml b/object_detection/src/config_file_examples/deployment_n6_yolov8_config.yaml
@@ -1,5 +1,5 @@
 general:
-  model_path: https://github.com/stm32-hotspot/ultralytics/blob/main/examples/YOLOv8-STEdgeAI/stedgeai_models/object_detection/yolov8n_256_quant_pc_uf_od_coco-person-st.tflite
+  model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/object_detection/yolov8n_256_quant_pc_uf_od_coco-person-st.tflite
   model_type: yolo_v8
 operation_mode: deployment
 
diff --git a/pose_estimation/deployment/README_MPU.md b/pose_estimation/deployment/README_MPU.md
@@ -24,7 +24,7 @@ To facilitate the deployment and avoid tools installation, the MPU deployment is
 
 You can use the deployment service by using a model zoo pre-trained model from the [STM32 model zoo](../pretrained_models/README.md) or your own pose estimation model. Please refer to the YAML file [deployment_mpu_config.yaml](../src/config_file_examples/deployment_mpu_config.yaml), which is a ready YAML file with all the necessary sections ready to be filled, or you can update the [user_config.yaml](../src/user_config.yaml) to use it.
 
-As an example, we will show how to deploy the model [yolov8n_256_quant_pt_uf_pose_coco-st.nb.tflite](../path/to/yolov8n_256_quant_pt_uf_pose_coco-st.nb.tflite) pre-trained on the COCO dataset. To use this model, you should first download it from the stm32-hotspot GitHub: https://github.com/stm32-hotspot/ultralytics
+As an example, we will show how to deploy the model **yolov8n_256_quant_pt_uf_pose_coco-st.nb.tflite** pre-trained on the COCO dataset. To use this model, you should first download it from the stm32-hotspot GitHub: https://github.com/stm32-hotspot/ultralytics
 
 </details></ul>
 </details>
diff --git a/pose_estimation/src/README.md b/pose_estimation/src/README.md
@@ -131,7 +131,7 @@ The `general` section and its attributes are shown below.
 general:
   project_name: COCO2017_pose_Demo  # Project name. Optional, defaults to "<unnamed>".
   model_type: yolo_mpe              # Name of the model 
-  model_path: <file-path>           # Path to a model file.
+  model_path: <file-path>           # Path to a model file or the raw link to the model from ultralytics.
   logs_dir: logs                    # Name of the directory where log files are saved. Optional, defaults to "logs".
   saved_models_dir: saved_models    # Name of the directory where model files are saved. Optional, defaults to "saved_models".
   deterministic_ops: False          # Enable/disable deterministic operations (a boolean). Optional, defaults to False.
diff --git a/pose_estimation/src/config_file_examples/deployment_n6_yolo_mpe_config.yaml b/pose_estimation/src/config_file_examples/deployment_n6_yolo_mpe_config.yaml
@@ -1,5 +1,5 @@
 general:
-   model_path: path/to/stm32-hotspot/ultralytics/yolov8n_256_quant_pt_uf_pose_coco-st.tflite
+   model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/pose_estimation/yolov8n_256_quant_pt_uf_pose_coco-st.tflite
    model_type: yolo_mpe
 
 operation_mode: deployment
diff --git a/pose_estimation/src/config_file_examples/evaluation_mpe_config.yaml b/pose_estimation/src/config_file_examples/evaluation_mpe_config.yaml
@@ -1,5 +1,5 @@
 general:
-   model_path: ../../../model_zoo/pose_estimation/yolov8n_pose/yolov8n_pose_256_int8.tflite
+   model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/pose_estimation/yolov8n_256_quant_pt_uf_pose_coco-st.tflite
    model_type: yolo_mpe
    num_threads_tflite: 8
 
diff --git a/pose_estimation/src/config_file_examples/prediction_mpe_config.yaml b/pose_estimation/src/config_file_examples/prediction_mpe_config.yaml
@@ -1,5 +1,5 @@
 general:
-   model_path: ../../../model_zoo/pose_estimation/yolov8n_pose/yolov8n_pose_256_int8.tflite
+   model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/pose_estimation/yolov8n_256_quant_pt_uf_pose_coco-st.tflite
    model_type: yolo_mpe  # spe, yolo_mpe
 
 operation_mode: prediction
diff --git a/pose_estimation/src/evaluation/README.md b/pose_estimation/src/evaluation/README.md
@@ -21,13 +21,13 @@ model using our evaluation service.
 <ul><details open><summary><a href="#1-1">1.1 Set the model and the operation mode</a></summary><a id="1-1"></a>
 
 As mentioned previously, all the sections of the YAML file must be set in accordance with
-this **[README.md](../config_file_examples/evaluation_config.yaml)**.
+this **[YAML file](../config_file_examples/evaluation_mpe_config.yaml)**.
 In particular, `operation_mode` should be set to evaluation and the `evaluation` section should be filled as in the
 following example:
 
 ```yaml
 general:
-  model_path: yolov8n_pose_int8.tflite
+  model_path: https://github.com/stm32-hotspot/ultralytics/raw/refs/heads/main/examples/YOLOv8-STEdgeAI/stedgeai_models/pose_estimation/yolov8n_256_quant_pt_uf_pose_coco-st.tflite
   model_type: yolo_mpe
 
 operation_mode: evaluation
