【YOLOv5】调用本地摄像头

YOLOv5源码:https://github.com/ultralytics/yolov5

注:本文用的是yolov5 v5.0版本,其他版本情况可能不同。
最近用YOLOv5做目标检测,直接调用本地摄像头会报错,需要在dataset中做一点修改。
具体如下:
【YOLOv5】调用本地摄像头_第1张图片
在279行的这两处改成str类型
【YOLOv5】调用本地摄像头_第2张图片
然后在detect里把这里的参数改为0
【YOLOv5】调用本地摄像头_第3张图片
然后运行detect.py即可调用本地摄像头。

参考链接:目标检测—教你利用yolov5训练自己的目标检测模型

你可能感兴趣的:(目标检测,深度学习,计算机视觉)