该项目是在下面连接的基础上实现的:
YOLOv5+DeepSort实现目标跟踪_yolov5目标跟踪_Albert_yeager的博客-CSDN博客
注意:文中的代码没有yolov5的模型文件,随便在github上找一个yolov5模型的源码,把utils和models文件下载放进去就可以了。(这里面放的是yolo模型的一些基本的文件)
随便找一个:https://github.com/giuliomattolin/yolov5_v5.0
然后创建一个weights文件目录,这里面就放自己训练的需要检查目标的权重文件。
该教程使用的官方的文件,运行的时候会直接下载,不过权重形状大小好像不匹配,我后面放了解决方法。
运行命令:
进行视频检测的命令:
python track.py --source 1.mp4 --save-vid --yolo_weights yolov5/weights/yolov5s.pt
使用摄像头进行检测的命令:(如果"q"无法退出,直接ctrl+c强制退出也可以)
python track.py --source 0 --show-vid --save-vid --yolo_weights yolov5/weights/yolov5s.pt
对运行过程中的三个问题进行总结:
1.Can't get attribute 'SPPF' on <module 'models.common' from...?
解决:
运行yolo v5-5.0版本报错找不到SPPF错误,进行解决_AI炮灰的博客-CSDN博客
2.The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3?
3.调用摄像头,报错:TypeError: argument of type 'int' is not iterable?
解决:
【目标检测】YOLOv5 5.0调用摄像头报错的处理_yolov5无法打开摄像头_Cul_accino的博客-CSDN博客