模型下载
https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt
安装库 ultralytics & opencv
pip install ultralytics opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
调用电脑camera验证YOLO模型
from ultralytics import YOLO
import cv2
# 加载模型
model = YOLO('yolov8n.pt') # 首次运行会自动下载模型(已离线下载,在.py同路径)
# 打开摄像头 (0 代表默认摄像头)
cap = cv2.VideoCapture(0)
# 如果你的摄像头是 USB 2.0 且卡顿,可以尝试强制设置 MJPG 编码[reference:15]
# cap.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter_fourcc('M', 'J', 'P', 'G'))
while cap.isOpened():
success, frame = cap.read()
if not success:
break
# 运行 YOLO 检测
results = model(frame, stream=True)
for r in results:
annotated_frame = r.plot()
cv2.imshow("YOLOv8 实时检测", annotated_frame)
if cv2.waitKey(1) & 0xFF == ord("q"):
break
cap.release()
cv2.destroyAllWindows()