1. github下载yolov7源码
https://github.com/WongKinYiu/yolov7
2. 修改bug
原程序调用电脑相机会报错,因此作如下修改:
找到yolov7-main\utils\datasets.py文件,找到285行
修改前:
if 'youtube.com/' in url or 'youtu.be/' in url: # if source is YouTube video
修改后:
if 'youtube.com/' in str(url) or 'youtu.be/' in str(url):
3. 打开命令行,进入到安装过pytorch的虚拟环境
4. cd到yolov7文件夹
输入
python detect.py --source 0
检测效果如下: