YOLOv5-第Y1周:调用官方权重进行检测
- YOLOv5-第Y1周:调用官方权重进行检测
- 一、前言
- 二、我的环境
- 三、下载源码
- 四、运行代码
- 五、视频检测
- 八、总结
YOLOv5-第Y1周:调用官方权重进行检测
一、前言
- 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
- 🍖 原作者:K同学啊
二、我的环境
- 电脑系统:Windows 10
- 语言环境:Python 3.8.5
- 编译器:colab在线编译
- 深度学习环境:PyTorch
三、下载源码
1、点击YOLOv5s开源地址下载YOLOv5s对应的开源代码
2、安装环境依赖包,下载上述开源代码后,打开cmd,进入对应的文件地址栏 输入pip install -r requirements.txt。例如我的文件地址为:C:\YOLOv5\yolov5-master\,则对应的命令为pip install -r C:\YOLOv5\yolov5-master\requirements.txt。在若速度过慢出现TimeoutError,就使用清华源的镜像下载,把命令修改为pip install -r C:\YOLOv5\yolov5-master\requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/。可参考解决yolov5s安装requirements.txt依赖库过慢出现timeout的问题这篇文章。
四、运行代码
以同样的方法打开cmd ,输入命令:python C:\YOLOv5\yolov5-master\detect.py --source C:\YOLOv5\yolov5-master\data\images\zidane.jpg --weights yolov5s.pt --img 640
在这个过程中,可能会一直出现错误,将K同学的yolov5s.pt(提取码为:bum6)文件放到项目的主目录下就ok了,上面的命令就是下载这个yolov5s.pt文件
五、视频检测
K同学的视频文件(提取码:9vaq).
下载视频文件后,
以同样的方法打开cmd输入命令:python C:\YOLOv5\yolov5-master\detect.py --source C:\YOLOv5\yolov5-master\MyVideo_2.mp4
最后一行是检测结果保存的路径。
进入打开就会得到如下的运行结果:
八、总结
本周开始进入yolov5的章节的学习。