examples: hailo: Switch default models for detect and pose demos

Use the hailo_architecture() helper to see which device is attached, and
select the appropriate HEF file for the architecture in use.

Signed-off-by: Naushir Patuck <naush@raspberrypi.com>

Author: naushir

examples/hailo/detect.py

@@ -5,9 +5,8 @@
 import argparse
 
 import cv2
-
 from picamera2 import MappedArray, Picamera2, Preview
-from picamera2.devices import Hailo
+from picamera2.devices import Hailo, hailo_architecture
 
 
 def extract_detections(hailo_output, w, h, class_names, threshold=0.5):
@@ -38,14 +37,22 @@ def draw_objects(request):
 if __name__ == "__main__":
     # Parse command-line arguments.
     parser = argparse.ArgumentParser(description="Detection Example")
-    parser.add_argument("-m", "--model", help="Path for the HEF model.",
-                        default="/usr/share/hailo-models/yolov8s_h8l.hef")
+    parser.add_argument("-m", "--model", help="Path for the HEF model. "
+                                              "Defaults to /usr/share/hailo-models/yolov8s_h8l.hef for H8 devices, "
+                                              "and /usr/share/hailo-models/yolov8m_h10.hef for H10 devices.",
+                        default=None)
     parser.add_argument("-l", "--labels", default="coco.txt",
                         help="Path to a text file containing labels.")
     parser.add_argument("-s", "--score_thresh", type=float, default=0.5,
                         help="Score threshold, must be a float between 0 and 1.")
     args = parser.parse_args()
 
+    if args.model is None:
+        if hailo_architecture() == 'HAILO10H':
+            args.model = '/usr/share/hailo-models/yolov8m_h10.hef'
+        else:
+            args.model = '/usr/share/hailo-models/yolov8s_h8l.hef'
+
     # Get the Hailo model, the input size it wants, and the size of our preview stream.
     with Hailo(args.model) as hailo:
         model_h, model_w, _ = hailo.get_input_shape()

examples/hailo/pose.py

@@ -3,15 +3,24 @@
 import argparse
 
 import cv2
-from pose_utils import postproc_yolov8_pose
-
 from picamera2 import MappedArray, Picamera2, Preview
-from picamera2.devices import Hailo
+from picamera2.devices import Hailo, hailo_architecture
+
+from pose_utils import postproc_yolov8_pose
 
 parser = argparse.ArgumentParser(description='Pose estimation using Hailo')
-parser.add_argument('-m', '--model', help="HEF file path", default="/usr/share/hailo-models/yolov8s_pose_h8l_pi.hef")
+parser.add_argument('-m', '--model', help="HEF file path"
+                                          "Defaults to /usr/share/hailo-models/yolov8s_pose_h8l_pi.hef for H8 devices, "
+                                          "and /usr/share/hailo-models/yolov8s_pose_h10.hef for H10 devices.",
+                    default=None)
 args = parser.parse_args()
 
+if args.model is None:
+    if hailo_architecture() == 'HAILO10H':
+        args.model = '/usr/share/hailo-models/yolov8s_pose_h10.hef'
+    else:
+        args.model = '/usr/share/hailo-models/yolov8s_pose_h8l_pi.hef'
+
 NOSE, L_EYE, R_EYE, L_EAR, R_EAR, L_SHOULDER, R_SHOULDER, L_ELBOW, R_ELBOW, \
     L_WRIST, R_WRIST, L_HIP, R_HIP, L_KNEE, R_KNEE, L_ANKLE, R_ANKLE = range(17)