yolo11 打开摄像头 实时检测

import cv2
from ultralytics import YOLO

# 加载预训练模型（选择适合的模型，如 yolov8n.pt, yolov8s.pt 等）
model = YOLO('cloth_best.pt')  # 会自动下载或使用本地模型

# 打开摄像头（0 表示默认摄像头）
cap = cv2.VideoCapture(0)

while cap.isOpened():
    # 读取摄像头画面
    success, frame = cap.read()
    
    if not success:
        print("无法读取摄像头画面")
        break
    
    # 使用YOLO进行检测
    results = model(frame, stream=True)  # stream=True 优化实时性能
    
    # 遍历检测结果
    for r in results:
        # 绘制检测框和标签
        annotated_frame = r.plot()  # 自动绘制结果
        
        # 显示带标注的画面
        cv2.imshow('YOLO Real-Time Detection', annotated_frame)
    
    # 按 'q' 退出循环
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

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