YOLOv5 调用wandb包,实现在启智社区设备上的的可视化模型训练。
首先进行wandb 的账号注册,点击wandb 进入,进入到页面后点击右上角 SIGN UP 进入到注册页面。
这里我们选择QQ邮箱的注册方式(因为国内QQ邮箱登录比较方便,GitHub以及谷歌邮箱什么的可能需要一些魔法),输入QQ邮箱账号以及密码后,回车。
然后进入邮箱认证环节,在QQ邮箱的收件箱中找到验证的电子邮件,进行验证。
验证完成后,进入创建账户页面,按要求填写即可(其实可以随便填写)。
账户创建完成后,进入到以下界面,复制可视化训练所需要的账户密钥。
至此,我们的前期准备工作算是完成。
这里的代码,我选用我自己上传的YOLOv5-6.1 官方版本
自动化脚本
早已配置明确,点击即用。
在此代码
第28行加入wandb的账户密钥。
然后在训练任务中,运行自动化训练脚本即可实现训练任务的可视化,训练完成后,模型以及本地产生的一切数据文件都会被上传到wandb云端。
答:(1)在Media项目栏中,你可以看到本地训练中的一切数据文件,如下:
(2)在Media项目栏中,你可以看到训练过程中的模型指标走势图。
(3)在train与val项目栏中,你可以看到训练过程中的模型损失指标走势图。
(4)学习率走势图。
(5)在System 项目栏中,你可以看到训练过程中显卡的利用率,温度,显存存占用率,内存占用率等等。
答: 点击右侧的项目名,进入到项目内部(项目名可以指定修改)
点击下图的中的标记按钮,即可找到模型文件。
进入到下图后,然后点击画圈的部分,即可找到存放模型的文件夹
进入到下图后,点击画圈部分,即可找到模型。
点击下载(下载后需要修改文件扩展名为 .pt)。
感谢平台给我提供的设备资源,祝愿社区平台越办越好😊。
``
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》