本文代码仅为示例,需自行修改。
此外,建议多看官方文档,也可看官方文档下的留言讨论,找自己需要的实现逻辑。现在ultralytics网站支持中文,阅读也不存在难度。
传送门:
ultralytics Yolov8官网
以下示例,采用OpenCV来从摄像头不断获取数据,调用Yolo模型推理。如有问题还望指出。
from ultralytics import YOLO
import cv2
model = YOLO('yolov8n.pt')
cap = cv2.VideoCapture(0) # 0表示调用摄像头
while True:
ret, frame = cap.read()
if not ret:
break
results = model(frame)
annotated_frame = results[0].plot()
cv2.imshow('YOLOv8', annotated_frame)
if cv2.waitKey(1) == ord('q'): # 按下q退出
break
cap.release()
cv2.destroyAllWindows()