青海营销网站建设服务图片展示 wordpress

本文将介绍如何使用本机摄像头进行目标检测实时推理的python代码。

一、下载YOLO权重文件

https://github.com/ultralytics/ultralytics?tab=readme-ov-file

拉到网页最下面，选择适合的模型，下载到本地。

二、环境配置

安装以下两个库，

pip install ultralytics
pip install opencv-python

三、完整代码

将yolov8n.pt保存在代码的同级目录下，当然你可以保存在任何地方，只需修改权重文件路径即可。

import cv2
from ultralytics import YOLO
import datetime# 加载YOLOv8模型
model = YOLO('./yolov8n.pt')# 打开默认摄像头 (设备索引为 0)
cap = cv2.VideoCapture(0)# 检查摄像头是否成功打开
if not cap.isOpened():print("无法打开摄像头")exit()while True:# 读取摄像头的每一帧ret, frame = cap.read()# 如果读取帧失败，则退出循环if not ret:print("无法接收帧（可能是摄像头断开）")break# 使用YOLOv8进行检测results = model(frame)# 解析结果并绘制检测框for result in results:boxes = result.boxesfor box in boxes:x1, y1, x2, y2 = map(int, box.xyxy[0])  # 获取检测框坐标confidence = box.conf[0]  # 获取置信度cls = int(box.cls[0])  # 获取类别label = model.names[cls]  # 获取类别名称# 绘制检测框和标签cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)cv2.putText(frame, f"{label} {confidence:.2f}", (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)# 获取当前时间now = datetime.datetime.now()time_str = now.strftime("%Y-%m-%d %H:%M:%S")cv2.putText(frame, time_str, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2, cv2.LINE_AA)# 显示当前帧cv2.imshow('Camera', frame)# 按下 'q' 键退出if cv2.waitKey(1) & 0xFF == ord('q'):break# 释放摄像头并关闭窗口
cap.release()
cv2.destroyAllWindows()