前两天有同学问Yolov5算法问题,想训练模型用于自己的应用,所以有了这篇文章,供需要的同学参考。
YOLOv5(You only Look once version 5)是一种基于深度学习的实时目标检测算法,由 Ultralytics 公司开发并开源。该系列算法因其速度快、准确度高而广受关注,在计算机视觉和人工智能应用领域,如视频监控、自动驾驶、机器人导航、图像检索等有着广泛的应用。
本系列教程将分为以下两个部分向大家详细介绍:
- 本地运行Yolov5算法
- 训练自己的检测模型
此篇文章将演示第一部分的操作过程。
首先我们得准备一些软件:
- Anaconda 包管理器
- Pycharm 编辑器
如果你还没有这两个软件,别担心,后台回复“安装包”,获取百度网盘下载地址。
1. 安装(推荐使用,便于安装包的管理):
安装包下载地址可选用清华源镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
选择对应操作系统的安装包下载后自行安装即可。本文以 Windows 系统为例,选择后缀文件下载。全部选择默认安装即可。
2. 获取源码
从 GitHub 网站下载 yolov5 源码:
https://github.com/ultralytics/yolov5
可使用如下 git 命令下载:
因为该网站服务器在国外,下载速度慢的同学可后台回复获取百度网盘下载地址。
解压压缩包后可获得如下文件目录:
3. 创建yolov5运行环境
使用conda命令创建名为yolov5的环境,进入开始菜单栏的所有应用列表
选择 Ananconda 目录下的 ,打开一个终端
输入如下 命令新建环境,并指定Python版本
输入进行安装,等待安装完成后输入如下命令激活 环境。
出现上图则表示进入环境成功。
4. 切换路径
进入到下载后的yolov5文件夹下,从终端打开,使用pip命令安装。从文件夹地址栏获取yolov5所在文件夹地址,然后在终端切换到该目录,以下图为例。
出现类似该图则表示地址切换成功,之后输入如下命令安装yolov5模型运行依赖。
以上命令使用阿里源镜像作为下载源,能够大幅度提升下载速度,建议使用。因为需要安装的工具包比较多,需要等待一定的时间。
出现下图所示结果,那么恭喜你,模型运行环境安装成功了!你距离最后成功仅一步之遥了。
1. 环境切换
点击Pycharm右下角的环境选择按钮切换环境。
如果新建环境不在列表中,则点击 Add New Interperter 按照下图选择yolov5环境。
2. 下载预训练模型并加载
从官网上下载yolov5s.pt模型文件(分享文件中已包含,不用另外下载),并在本地的yolov5文件夹下创建checkpoints文件夹,再将下载的模型文件放入其中,如下图所示。
然后再在yolov5目录下新建一个test.py文件,打开该文件后输入如下代码:
3. 图片推理
最后会弹出一张图片,如下所示,图中红色框即为yolov5s模型检测到的目标物体。
自此,本教程第一部分完毕,请期待下一篇训练自己的yolov5模型教程吧~
更多精彩文章请关注微信公众号:手把手PythonAI编程