|
|
@@ -0,0 +1,91 @@ |
|
|
|
## YOLOv5可视化训练 |
|
|
|
|
|
|
|
### 分享者:[a775929656a](https://openi.pcl.ac.cn/a775929656a) |
|
|
|
|
|
|
|
|
|
|
|
### 前言: |
|
|
|
YOLOv5 调用wandb包,实现在启智社区设备上的的可视化模型训练。 |
|
|
|
|
|
|
|
|
|
|
|
### 准备工作: |
|
|
|
首先进行wandb 的账号注册,点击[wandb](https://wandb.ai/) 进入,进入到页面后点击右上角 SIGN UP 进入到注册页面。 |
|
|
|
|
|
|
|
<img src= img/z-0218-1.png> |
|
|
|
|
|
|
|
|
|
|
|
这里我们选择QQ邮箱的注册方式(因为国内QQ邮箱登录比较方便,GitHub以及谷歌邮箱什么的可能需要一些魔法),输入QQ邮箱账号以及密码后,回车。 |
|
|
|
|
|
|
|
<img src= img/z-0218-2.png> |
|
|
|
|
|
|
|
然后进入邮箱认证环节,在QQ邮箱的收件箱中找到验证的电子邮件,进行验证。 |
|
|
|
|
|
|
|
<img src= img/z-0218-3.png> |
|
|
|
|
|
|
|
验证完成后,进入创建账户页面,按要求填写即可(其实可以随便填写)。 |
|
|
|
|
|
|
|
<img src= img/z-0218-4.png> |
|
|
|
|
|
|
|
账户创建完成后,进入到以下界面,复制可视化训练所需要的账户密钥。 |
|
|
|
|
|
|
|
<img src= img/z-0218-5.png> |
|
|
|
|
|
|
|
至此,我们的前期准备工作算是完成。 |
|
|
|
|
|
|
|
### YOLOv5 代码可视化设置: |
|
|
|
这里的代码,我选用我自己上传的[YOLOv5-6.1 官方版本](https://openi.pcl.ac.cn/a775929656a/2022-9-8) |
|
|
|
[自动化脚本](https://openi.pcl.ac.cn/a775929656a/2022-9-8/src/branch/master/main.py) |
|
|
|
早已配置明确,点击即用。 |
|
|
|
在此[代码](https://openi.pcl.ac.cn/a775929656a/2022-9-8/src/branch/master/utils/loggers/__init__.py) |
|
|
|
第28行加入wandb的账户密钥。 |
|
|
|
|
|
|
|
<img src= img/z-0218-6.png> |
|
|
|
|
|
|
|
然后在训练任务中,运行自动化训练脚本即可实现训练任务的可视化,训练完成后,模型以及本地产生的一切数据文件都会被上传到wandb云端。 |
|
|
|
|
|
|
|
### YOLOv5 训练可视化都可以看到数据? |
|
|
|
答:(1)在Media项目栏中,你可以看到本地训练中的一切数据文件,如下: |
|
|
|
|
|
|
|
<img src= img/z-0218-7.png> |
|
|
|
|
|
|
|
(2)在Media项目栏中,你可以看到训练过程中的模型指标走势图。 |
|
|
|
|
|
|
|
<img src= img/z-0218-8.png> |
|
|
|
|
|
|
|
(3)在train与val项目栏中,你可以看到训练过程中的模型损失指标走势图。 |
|
|
|
<img src= img/z-0218-9.png> |
|
|
|
|
|
|
|
(4)学习率走势图。 |
|
|
|
|
|
|
|
<img src= img/z-0218-10.png> |
|
|
|
|
|
|
|
(5)在System 项目栏中,你可以看到训练过程中显卡的利用率,温度,显存存占用率,内存占用率等等。 |
|
|
|
|
|
|
|
<img src= img/z-0218-11.png> |
|
|
|
|
|
|
|
### 训练完成后如何找到训练所产生的模型文件? |
|
|
|
答: 点击右侧的项目名,进入到项目内部(项目名可以指定修改) |
|
|
|
|
|
|
|
<img src= img/z-0218-12.png> |
|
|
|
|
|
|
|
<img src= img/z-0218-13.png> |
|
|
|
|
|
|
|
点击下图的中的标记按钮,即可找到模型文件。 |
|
|
|
|
|
|
|
<img src= img/z-0218-14.png> |
|
|
|
|
|
|
|
进入到下图后,然后点击画圈的部分,即可找到存放模型的文件夹 |
|
|
|
|
|
|
|
<img src= img/z-0218-15.png> |
|
|
|
|
|
|
|
进入到下图后,点击画圈部分,即可找到模型。 |
|
|
|
|
|
|
|
<img src= img/z-0218-16.png> |
|
|
|
|
|
|
|
点击下载(下载后需要修改文件扩展名为 .pt)。 |
|
|
|
|
|
|
|
<img src= img/z-0218-17.png> |
|
|
|
|
|
|
|
### 致谢 |
|
|
|
感谢平台给我提供的设备资源,祝愿社区平台越办越好😊。 |
|
|
|
|
|
|
|
`` |