【YOLO】数据增强bug

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

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

问题分析

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

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

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

问题解决

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

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

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

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

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

原始图像

相关推荐
程序员JerrySUN5 小时前
RK3588 项目实战总结:从 GStreamer、DP 显示、OpenCV 到 YOLO 部署的一次完整梳理
人工智能·opencv·yolo
Cc_Debugger6 小时前
【饿了么plus-table】开启多选时,点击下面的单选按钮,页面显示是全选的样子,bug
bug
Byron Loong6 小时前
【机器视觉】标注软件 labelMe和lableImg对比
yolo·计算机视觉
QQ676580087 小时前
AI赋能识别之围栏破损识别 围栏缺陷检测数据集 栅栏破损识别数据集围栏孔洞识别植被入侵检测图像数据集 目标检测图像数据集第10125期
人工智能·yolo·目标检测·目标跟踪·围栏破损识别·围栏缺陷识别·栅栏破损识别
龙卷风卷云7 小时前
【BUG】Nginx使用upstream后端接口报 400
运维·nginx·bug
深度学习lover8 小时前
<数据集>yolo扑克牌识别<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·扑克牌识别
QQ676580089 小时前
AI人工智能图像识别 兔子动物分类研究 宠物行业物种鉴别及畜牧业兔种监测 兔种监测识别 YOLO图像数据集 兔类物种的计算机视觉识别模型训练 第10363期
人工智能·yolo·目标检测·目标跟踪·分类·宠物·宠物行业鉴别
melonbo1 天前
Jetson Nano 平台YOLO目标检测优化详细设计方案
yolo·目标检测·目标跟踪
Linux猿2 天前
植物病害图像数据集 YOLO 目标检测 | 可下载
人工智能·yolo·目标检测·yolo数据集·yolo目标检测·yolo目标检测数据集·植物病害图像数据集
毕胜客源码2 天前
改进yolov8的香蕉成熟度检测系统,改进前后的模型指标对比,有技术文档,支持图像、视频和摄像实时检测
人工智能·python·深度学习·yolo·django