【YOLO】数据增强bug

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

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

问题分析

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

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

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

问题解决

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

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

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

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

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

原始图像

相关推荐
fl17683115 小时前
无人机视角低空拍摄的土豆马铃薯幼苗与杂草检测数据集VOC+YOLO格式5266张6类
yolo·无人机
qq_25294131681 天前
列车车轮缺陷智能检测数据集:800张图像、4大类别,助力铁路安全运维
运维·人工智能·安全·yolo·目标检测·计算机视觉·视觉检测
雨晨源码(同名B站)1 天前
基于深度学习YoloV11农业病害虫害检测系统 智慧农业信息化综合管理平台 (附源码+lw文档+ppt)
数据库·人工智能·深度学习·yolo·信息可视化
深念Y1 天前
AMD 芯片组驱动触发高频 SSD 写入的问题及解决方案
windows·bug·ssd·日志·芯片·驱动·amd
加油码2 天前
探秘 volatile关键字:编译器的过度优化
linux·bug·信号处理
鹿屿二向箔2 天前
keil5.39版本使用AC6编译器不能位域操作的BUG使用STM32U385系列MCU
stm32·单片机·bug
angryshan2 天前
goLang配置断点流程
go·bug
guo_xiao_xiao_2 天前
YOLO工业质检与产品展示灯泡目标检测数据集-509张
yolo·目标检测·目标跟踪
探物 AI3 天前
yolo检测中的激活函数19:ReLU激活函数 (Rectified Linear Unit)
网络·人工智能·深度学习·yolo
guo_xiao_xiao_3 天前
YOLO人物肖像发型分类目标检测数据集
yolo·目标检测·分类