小编Lou*_*uis的帖子

我想将 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()

我想在 OpenCV 中显示 YOLO 带注释的图像。我知道我可以在model.predict(source='0', show=True). 但我想连续监视我的程序的预测类名称，同时显示图像输出。