基于YOLOv10的摄像头实时检测

本项目基于开源项目：https://github.com/THU-MIG/yolov10?tab=readme-ov-file
实现一个简单的实时检测效果。

Start capturing from the webcam

import cv2

cap = cv2.VideoCapture(0) # 从默认摄像头获取数据
cap.set(3, 640)
cap.set(4, 480) # 设置显示图像的长和宽

while True:
    ret, img= cap.read()
    cv2.imshow('Webcam', img) # 显示图像

    if cv2.waitKey(1) == ord('q'):
        break # 设置退出条件

cap.release()
cv2.destroyAllWindows()

Complete code — object detection with YOLO and webcam

import cv2
from ultralytics import YOLOv10 

# 模型路径和配置
image_size = 640
conf_threshold = 0.25
model_path = '/yolov10-main/yolov10b.pt'

# 加载模型
model = YOLOv10(model_path)

# 打开摄像头
cap = cv2.VideoCapture(0)
cap.set(3, 640)
cap.set(4, 480)

while True:
    ret, img = cap.read()
    if not ret:
        break

    # 使用模型进行预测
    results = model.predict(source=img, imgsz=image_size, conf=conf_threshold, save=False)
    
    # 获取标注后的图像
    annotated_img = results[0].plot()
    
    # 显示结果
    cv2.imshow('YOLOv10 Real Time Dectection', annotated_img)
    if cv2.waitKey(1) == ord('q'):
        break

# 释放资源
cap.release()
cv2.destroyAllWindows()

这是一个简单版本，更精美的方法可以用gradio实现。

效果

参考文献
https://dipankarmedh1.medium.com/real-time-object-detection-with-yolo-and-webcam-enhancing-your-computer-vision-skills-861b97c78993