Update: class lables

yolov5.py

@@ -27,13 +27,16 @@ def parse_detections(results):
     return boxes, colors, names
 
 
-def draw_detections(boxes, colors, names, img):
-    for box, color, name in zip(boxes, colors, names):
+def draw_detections(boxes, colors, names, classes, img):
+    for box, color, name, cls in zip(boxes, colors, names, classes):
         xmin, ymin, xmax, ymax = box
+        label = f"{cls}: {name}"  # Combine class ID and name
         cv2.rectangle(img, (xmin, ymin), (xmax, ymax), color, 2)
-        cv2.putText(img, name, (xmin, ymin - 5),
-                    cv2.FONT_HERSHEY_SIMPLEX, 0.8, color, 2,
-                    lineType=cv2.LINE_AA)
+        cv2.putText(
+            img, label, (xmin, ymin - 5),
+            cv2.FONT_HERSHEY_SIMPLEX, 0.8, color, 2,
+            lineType=cv2.LINE_AA
+        )
     return img
 
 

yolov8.py

@@ -24,13 +24,16 @@ def parse_detections(detections, model):
         names.append(name)
     return boxes, colors, names
 
-def draw_detections(boxes, colors, names, img):
-    for box, color, name in zip(boxes, colors, names):
+def draw_detections(boxes, colors, names, classes, img):
+    for box, color, name, cls in zip(boxes, colors, names, classes):
         xmin, ymin, xmax, ymax = box
+        label = f"{cls}: {name}"  # Combine class ID and name
         cv2.rectangle(img, (xmin, ymin), (xmax, ymax), color, 2)
-        cv2.putText(img, name, (xmin, ymin - 5),
-                    cv2.FONT_HERSHEY_SIMPLEX, 0.8, color, 2,
-                    lineType=cv2.LINE_AA)
+        cv2.putText(
+            img, label, (xmin, ymin - 5),
+            cv2.FONT_HERSHEY_SIMPLEX, 0.8, color, 2,
+            lineType=cv2.LINE_AA
+        )
     return img
 
 def generate_cam_image(model, target_layers, tensor, rgb_img, boxes):