【目标检测算法】利用wandb可视化YOLO-V5模型的训练

最新推荐文章于 2024-04-17 10:10:00 发布

三木今天学习了嘛

最新推荐文章于 2024-04-17 10:10:00 发布

阅读量3.9k

点赞数 13

分类专栏： # 目标检测文章标签：目标检测算法深度学习

本文链接：https://blog.csdn.net/weixin_45751396/article/details/126723357

版权

目标检测专栏收录该内容

7 篇文章 8 订阅

订阅专栏

文章目录

前言
一、wandb可视化模型训练
二、colab&wandb在线训练与在线参数视察

还是基于该博客做的wandb可视化模型的训练。
【目标检测算法】YOLO-V5实战检测VOC2007数据集

前言

训练过yolov5网络的都知道，网络会自动生成一个run文件夹，里面涵盖了几乎所有评价指标，为什么还要用wandb进行可视化训练，多此一举嘛不是。
这里放一个官网的图：

在这里插入图片描述

可以看到，wandb可以实现不同网络的对比寻优。当然这只是显眼易见的优点，肯定还有很多功能，我也没有发现。

一、wandb可视化模型训练

1. 注册wandb账号

Weights & Biases网上有很多博主教如何注册，这里就不再赘述。

2. 在pycharm配置wandb

在这里插入图片描述

点击这个Quickstart，官方有教程教你如何快速新建一个可视化工程。在这里插入图片描述
在pycharm输入pip install wandb–>安装完成输入wandb login–>复制下面的key输入pycharm

注意：输入wandb login后，光标将不存在，直接粘贴按回车就行

3. 调整代码，并在控制台启动wandb

这里考虑到GPU性能，我只跑了10轮。实际训练要训练至少50轮。batch size也调整为8，不然我的显存就爆了。
并且我把训练时输入图片大小调整为320*320，也是为了加快训练速度。（==没办法，GPU太垃圾了）

    parser.add_argument('--epochs', type=int, default=10, help='total training epochs')
    parser.add_argument('--batch-size', type=int, default=8, help='total batch size for all GPUs, -1 for autobatch')
    parser.add_argument('--imgsz', '--img', '--img-size', type=int, default=320, help='train, val image size (pixels)')

如果你是第一次，下面回让你输入key，直接复制wandb的key粘贴就好了。这里我是第二次进入，就不需要输入啦。
在这里插入图片描述

4. 在wandb中查看训练过程

wandb: Tracking run with wandb version 0.13.2
wandb: Run data is saved locally in P:\Project_Python\yolov5-voc\wandb\run-20220906_082929-owk5vzjl
wandb: Run `wandb offline` to turn off syncing.
wandb: Syncing run legendary-hill-2
wandb:  View project at https://wandb.ai/sanmumumu/train
wandb:  View run at https://wandb.ai/sanmumumu/train/runs/owk5vzjl
YOLOv5 temporarily requires wandb version 0.12.10 or below. Some features may not work as expected.