yolov5代码链接:Github
本文的精华:YOLOv5训练权重
链接:https://pan.baidu.com/s/1n_p3Z7pGNdVYgdbJLzK70g
提取码:gyr9
YOLOv5文件
链接:https://pan.baidu.com/s/1LVkYlXJ9upnGvJ3VLA00Gg
提取码:r8tc
环境配置:
python3.5.2(可用)、python3.6、python3.7均可
GPU 、CPU、ubuntu、windows均可
torch>=1.4.0 & torchvision>=0.5.0
torch & torchvision(CPU)安装命令:
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
下载权重,配置环境,执行命令:
python detect.py #检测inference\images文件夹下的所有图片
python .\detect.py --source .\inference\images\bus.jpg # 检测某一张图片
python .\detect.py --source .\road.mp4 # 检测某一视频
python .\detect.py --source 0 # 调用摄像头
检测效果: