我想将OpenCV与来自“ultralytics”的YOLOv8集成,因此我想从模型预测中获取边界框坐标。我该如何做到这一点?
我想在OpenCV中显示YOLO注释图像。我知道我可以在
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)
中使用流参数。但是我想要连续监视预测的类名,同时显示图像输出。
model.predict
的输出吗? - Louisr.plot()
或results.plot()
呢?根据文档,它们应该可以工作,但实际上并没有。 - undefined