【YOLO】数据增强bug

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

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

问题分析

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

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

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

问题解决

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

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

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

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

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

原始图像

相关推荐
spencer_tseng2 小时前
Lombok.jar bug
bug·jar·lombok
B站_计算机毕业设计之家9 小时前
深度学习:Yolo水果检测识别系统 深度学习算法 pyqt界面 训练集测试集 深度学习 数据库 大数据 (建议收藏)✅
数据库·人工智能·python·深度学习·算法·yolo·pyqt
B站_计算机毕业设计之家13 小时前
深度学习:python人脸表情识别系统 情绪识别系统 深度学习 神经网络CNN算法 ✅
python·深度学习·神经网络·算法·yolo·机器学习·cnn
城南皮卡丘15 小时前
【数据集+源码】基于yolov11+streamlit的玉米叶片病虫害检测系统
yolo·目标检测·计算机视觉·智慧农业·玉米病害检测
王哈哈^_^16 小时前
【完整源码+数据集】蓝莓数据集,yolo11蓝莓成熟度检测数据集 3023 张,蓝莓成熟度数据集,目标检测蓝莓识别算法系统实战教程
人工智能·算法·yolo·目标检测·计算机视觉·ai·视觉检测
盘古开天166616 小时前
通俗易懂:YOLO模型原理详解,从零开始理解目标检测
人工智能·yolo·目标检测
王哈哈^_^16 小时前
【完整源码+数据集】高空作业数据集,yolo高空作业检测数据集 2076 张,人员高空作业数据集,目标检测高空作业识别系统实战教程
人工智能·算法·yolo·目标检测·计算机视觉·目标跟踪·视觉检测
海蓝可知天湛16 小时前
Ubuntu24.10禁用该源...+vmware无法复制黏贴“天坑闭环”——从 DNS 诡异解析到 Ubuntu EOL 引发的 apt 404排除折腾记
linux·服务器·安全·ubuntu·aigc·bug
AI浩1 天前
PAB-Mamba-YoLo: VSSM 辅助 YOLO 用于断奶仔猪攻击行为检测
yolo
程序员小远2 天前
快速定位bug,编写测试用例
自动化测试·软件测试·python·测试工具·职场和发展·测试用例·bug