Smikkelbakje/J2S1-Kobuki
1
0
Fork 0
mirror of https://gitlab.fdmci.hva.nl/technische-informatica-sm3/ti-projectten/rooziinuubii79.git synced 2025-08-04 04:14:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update yolo naming in image

Browse Source
This commit is contained in:
Sam Hos
2025-01-06 15:55:24 +01:00
parent 20d6d8799d
commit 7845feb9f8
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/Python/flask/web/app.py
View File

@@ -31,14 +31,14 @@ def on_message(client, userdata, message):
for result in results: for result in results:
for box in result.boxes: for box in result.boxes:
yolo_results.append({ yolo_results.append({
"class": yolo_classes, "class": yolo_classes[box.cls.item()],
"confidence": box.conf.item(), "confidence": box.conf.item(),
"bbox": box.xyxy.tolist() "bbox": box.xyxy.tolist()
}) })
# Draw bounding box on the image # Draw bounding box on the image
x1, y1, x2, y2 = map(int, box.xyxy[0]) x1, y1, x2, y2 = map(int, box.xyxy[0])
cv2.rectangle(latest_image, (x1, y1), (x2, y2), (0, 255, 0), 2) 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 # Create an MQTT client instance
mqtt_client = mqtt.Client() mqtt_client = mqtt.Client()