YOLOv8 получает прогнозируемую ограничивающую рамку ⇐ Python

[Anonymous]

Я хочу интегрировать OpenCV с YOLOv8 из ultralytics, поэтому я хочу получить координаты ограничивающего прямоугольника из прогноза модели. Как мне это сделать?

Код: Выделить всё

from ultralytics import YOLO
import cv2

model = YOLO('yolov8n.pt')
cap = cv2.VideoCapture(0)
cap.set(3, 640)
cap.set(4, 480)

while True:
_, frame = cap.read()

img = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

results = model.predict(img)

for r in results:
for c in r.boxes.cls:
print(model.names[int(c)])

cv2.imshow('YOLO V8 Detection', frame)
if cv2.waitKey(1) & 0xFF == ord(' '):
break

cap.release()
cv2.destroyAllWindows()

Я хочу отобразить аннотированное изображение YOLO в OpenCV. Я знаю, что могу использовать параметр потока в model.predict(source='0', show=True). Но я хочу постоянно отслеживать предсказанные имена классов для моей программы, одновременно отображая выводимые изображения.

Подробнее здесь: https://stackoverflow.com/questions/753 ... unding-box