From 7845feb9f8b2da32b81528b9272c59d48ecfc359 Mon Sep 17 00:00:00 2001 From: Sam Hos Date: Mon, 6 Jan 2025 15:55:24 +0100 Subject: [PATCH] update yolo naming in image --- src/Python/flask/web/app.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/Python/flask/web/app.py b/src/Python/flask/web/app.py index 0097c45..1a66899 100644 --- a/src/Python/flask/web/app.py +++ b/src/Python/flask/web/app.py @@ -31,14 +31,14 @@ def on_message(client, userdata, message): for result in results: for box in result.boxes: yolo_results.append({ - "class": yolo_classes, + "class": yolo_classes[box.cls.item()], "confidence": box.conf.item(), "bbox": box.xyxy.tolist() }) # Draw bounding box on the image x1, y1, x2, y2 = map(int, box.xyxy[0]) cv2.rectangle(latest_image, (x1, y1), (x2, y2), (0, 255, 0), 2) - cv2.putText(latest_image, f"{yolo_classes} {box.conf.item():.2f}", (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2) + cv2.putText(latest_image, f"{yolo_classes[box.cls.item()]} {box.conf.item():.2f}", (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2) # Create an MQTT client instance mqtt_client = mqtt.Client()