我想从ultralytics
集成OpenCV和YOLOv8,所以我想从模型预测中获得边界框坐标,我该怎么做?
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)
中使用stream参数,但是我想在显示图像输出的同时,持续监视程序的预测类名。
2条答案
按热度按时间06odsfpq1#
这将在原始RGB
frame
上绘制带有各自标签名称的bbox,而不是输入模型的BGRimg
:ffvjumwh2#
您可以使用下面的代码获取所有信息:
您可以在documentation中进一步阅读。