yolo-tiny可视化:loss和IOU波动

一位AI算法标注工程师分享了使用Yolo-tiny网络进行垃圾分类任务的训练过程,发现训练结果波动较大,疑似由数据集质量和learningrate设置不当引起。文章探讨了可能的解决方案,包括调整学习率、batch大小和subdivisions参数。
部署运行你感兴趣的模型镜像

在这里插入图片描述
在这里插入图片描述
这是我训练一个垃圾分类的yolo-tiny网络的输出可视化结果,怀疑可能有如下几条原因导致了如此波动:
1、数据集问题。大部分图片都是从网络上爬取的,少部分是自己拍摄的,质量层次不齐且无法充分表现出此类垃圾特征。因为已经检查过直接用来训练的txt文件,所以可以基本排除是格式转化出现的问题。
2、learning rate选择不合适。

batch=64
subdivisions=8
width=416
height=416
channels=3
momentum=0.9
decay=0.0005

angle=45
saturation = 3
exposure = 3
hue=.2

learning_rate=0.01
burn_in=1000
max_batches = 2500
policy=steps
steps=100,500,1000
scales=.1,.1,.1

这是我训练的参数,可能我的learning rate太大了,下次训练时准备调整learning_rate和steps以及steps。
或许可以尝试一下调整batch、subdivisions,增大batch或者减小sub(每batch/sub个样本更新一次参数。)

欢迎大家给小白提建议,以改善训练效果。

您可能感兴趣的与本文相关的镜像

Yolo-v8.3

Yolo-v8.3

Yolo

YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的Joseph Redmon 和Ali Farhadi 开发。 YOLO 于2015 年推出,因其高速和高精度而广受欢迎

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值