Adapting to Ultralyrics 8.0.43+ changed API

app.py

@@ -6,21 +6,24 @@ def inference(path:str, threshold:float=0.6):
     print("trying inference with path", path)
     if path is None:
         return None,0
-    model = YOLO('yolov8m.pt')
-
-    model.classes = [0]  # only considering class 'person' and not the 79 other classes...
+    model = YOLO('yolov8m.pt') # Caution new API since Ultralytics 8.0.43
+    # [0]  # only considering class 'person' and not the 79 other classes...
+    outputs = model.predict(source=path, classes= [0], show=False, conf=threshold) # new API with Ultralytics 8.0.43. Accepts 'show', 'classes', 'stream', 'conf' (default is 0.25)
     image = cv2.imread(path)
-    outputs = model.predict(source=path, return_outputs=True)
+    counter = 0
     for output in outputs:  # mono item batch
-        detections = output['det']
-        counter=0
-        for detection in detections:
-            if detection[4]<threshold:
+        conf = output.boxes.conf  # the tensor of detection confidences
+        xyxy = output.boxes.xyxy
+        cls = output.boxes.cls # 0 is 'person' and 5 is 'bus' 16 is dog
+        nb=cls.size(dim=0)
+        for i in range(nb):
+            box = xyxy[i]
+            if conf[i]<threshold:
                 break
             cv2.rectangle(
                 image,
-                (int(detection[0]), int(detection[1])),
-                (int(detection[2]), int(detection[3])),
+                (int(box[0]), int(box[1])),
+                (int(box[2]), int(box[3])),
                 color=(0, 0, 255),
                 thickness=2,
             )
@@ -32,7 +35,7 @@ gr.Interface(
     inputs = [ gr.components.Image(type="filepath", label="Input"), gr.Slider(minimum=0.5, maximum=0.9, step=0.05, value=0.7, label="Confidence threshold") ],
     outputs = [ gr.components.Image(type="numpy", label="Output"), gr.Label(label="nb of persons detected for given confidence threshold") ],
     title="Person detection with YOLO v8",
-    description="Person detection, you can tweak the corresponding confidence threshold. Good results even when face not visible.",
+    description="Person detection, you can tweak the corresponding confidence threshold. Good results even when face not visible. New API since Ultralytics 8.0.43.",
     examples=[ ['data/businessmen-612.jpg'], ['data/businessmen-back.jpg']],
     allow_flagging="never"
 ).launch(debug=True, enable_queue=True)