【YOLO】数据增强bug

最近在使用yolo。在mosaic增强时,发现了一些问题,这些问题可能严重影响训练。正准备提bug,记录一下。在我的场景中,根据yolo展示的训练截图,大约每9张就会出现一个严重错误。

由于我的目标是"斜跨标注框对角线"的,因此当某些标注框被截取的时候,非常有可能把标注框中,完全不含目标物的区域截出来,严重干扰模型训练。

问题分析

原始图像如下,标注比较合理,识别目标是绝缘子。

马赛克增强后的效果如下(我选了一个还能看懂的,实际情况中,有非常多特别离谱的错误)。可以看到,右下角的0标签,出现了明显的问题,比真实目标宽了很多,应该会影响到模型的训练。

类似的问题还包括,下图的情况,目标被剪裁到的只剩下了一个非常小的角,根本无法识别。类似的情况也比较多。

问题解决

当关闭马赛克后,类似的情况就少了很多(几乎没有),因此应该是马赛克过于粗暴了。

由于我的目标是斜跨标注框对角线的,因此在马赛克截取的时候,需要重新根据对角线位置计算新的坐标框(计算截取位置与对角线的交点)。

或者在我的场景下,不要使用mosaic增强,或者保证马赛克增强的时候,把某些特殊类别的坐标框完整的保留。

或者少用一些马赛克,提前关闭马赛克,但是未关闭的时候,造成的模型误会和训练浪费,恐怕难以纠正。

或者当马赛克将某个目标截取了超过50%面积的时候,自动将目标剩余的部分涂黑。

原始图像

相关推荐
梦之美丽生活2 小时前
【本地部署及docker部署】yolov8_detect
yolo·docker·容器
YOLO数据集集合16 小时前
基准标记目标检测数据集 |基准标记 视觉定位 相机标定 目标检测 YOLO格式 深度学习数据集 计算机9038期
数码相机·yolo·目标检测·视觉定位·基准标定
YOLO数据集集合19 小时前
隧道病害检测数据集 | 隧道检测 裂缝识别 渗水监测 剥落检测9037期
人工智能·深度学习·yolo·隧道·裂缝检测·裂缝识别·隧道病害
深瞳智检1 天前
深瞳智检(小程序)正式上线!【免费/快捷/方便】获取YOLO检测数据集
人工智能·yolo·目标检测·小程序·数据集·cv数据收集
a1117761 天前
SOP行为监控系统 开源项目yolo
yolo
深念Y1 天前
fnOS-k3s-ResolvConf权限700-Bug分析
内核·bug·nas·k3s·fnos·魔改内核
jay神1 天前
2026年YOLO还有哪些创新点可以做?
python·yolo·毕业设计·科研·课程设计·创新
不正经学生1 天前
C语言预处理详解:编译器真正动手之前的那些事
c语言·开发语言·算法·面试·bug
毕竟是shy哥1 天前
计算YOLO数据集中每个类的目标数
算法·yolo·机器学习
仙女修炼史2 天前
gradcam在yolo中的应用
人工智能·python·yolo