写这篇文章最大的初衷就是最近频繁的有很多人私信问我相关的数据集的问题,基本上都是从我前面的目标检测专栏里面的这篇文章过来的,感兴趣的话可以看下:
《轻量级模型YOLOv5-Lite基于自己的数据集【焊接质量检测】从零构建模型超详细教程》
保姆级的教程,小白即可直接上手操作实践整个完整流程。
很多人在最开始做目标检测的时候没有自己的数据集,或者是由于自己的专业方向需要的就是比如:缺陷、质量等特定专业领域内的数据集,所以看到这里就感觉很相关。
由于之前的文章过去的时间有点久了,上周一直也没时间去搜索和梳理相关的内容。
今天正好有点时间我重新冲数据库中检索出来了焊缝质量检测最原始的数据集,我们称之为【基础数据集】,如下所示:
JPEGImages目录如下:
labels目录如下:
xmls目录如下:
最原始批次的数据集共有1134个样本。
做CV相关任务我们应该都懂得这1k左右的数据量对于训练一个目标检测模型来说实在是有点捉襟见肘,所以从数据源头获取到更多的高质量的数据集显得就尤为重要了,基于对原始数据集的观察,我发现这里的采集的数据大都是方方正正的,我们可以先设计基础的方法进行扩充增强处理,就可以从源头端获取到更多的数据集了,这里我采取的方式包:左右倒置、上下置换、90°、180°和270°旋转处理,一共获取到了5670个样本数据,我们称之为【扩充数据集】,如下所示:
JPEGImages目录如下所示:
labels目录如下所示:
xmls目录如下所示:
完成这部分工作后同事提议说,也可以基于连续角度的旋转来构建广度更大的增强数据集,简答来说这里我们以15°作为最基本的角度间隔单元,从15°到300°,一共划分出来了17个单元,共生成了19278个新的样本数据,我们称之为【角度数据集】,如下所示:
JPEGImages目录如下所示:
labels目录如下所示:
xmls目录如下所示:
到这里,其实经过我们的一系列处理后就已经得到了基本够用于模型训练的数据集了,联想到之前我们项目里面实现和应用的一些增强的方法,这里我们又基于基础数据集来进行了一波增强处理,一共获得了12000个样本数据,我们称之为【增强数据集】,在这批增强数据里面主要包含三种常见的技术增强手段,分别为:随机增强处理、mixup增强处理、mosaic增强处理,每种技术增强均产生4000个样本数据,一共就是12000个样本数据,如下所示:
mixup目录如下所示:
JPEGImages目录如下所示:
xmls目录如下所示:
mosaic目录如下所示:
JPEGImages目录如下所示:
xmls目录如下所示:
random目录如下所示:
JPEGImages目录如下所示:
xmls目录如下所示:
整体数据情况如下:
在后续的开发工作中,因为实际项目复杂场景的客观存在,leader要求增加数据的丰富性,这里基于图像变换操作开发了新的增强方法,实现了新批量数据的扩充增强处理,这里称之为【变换增强数据集】
详情如下:
该批次增强得到的数据集与原始数据集较为相近,可用度还是很高的,适应了不同尺度的变化,共获取了将近1.6w的数据量。
之后leader基于aug增强方式,实现了更为复杂的增强扩充,这里新生成的图像数据已经是变化很大的了,相对原始数据来说差异性也更为明显,主要是想要生成困难识别的样本,让模型训练在实验数据的场景下能够更加鲁棒稳定,这里称之为【深度增强数据集】。
详情如下:
这批次共得到11429张图像数据,生成这批图像的难点在于得到图像复杂变化后与之对应的目标框的新的坐标位置。
数据整体详情如下:
【基础数据集】
1134
【扩充数据集】
5670
【角度数据集】
19278
【增强数据集】
12000
【变换增强数据集】
15936
【深度增强数据集】
11429
总计: 1134+5670+19278+12000+15936+11429=65447
数据开发制作不易,相互理解。
由于过去时间久远,部分数据集已经遗失,本身获取链接是都可以放在这里的,但是平台一直不给审核通过,实属无奈,如果有需要就私信联系我吧。
版权归原作者 Together_CZ 所有, 如有侵权,请联系我们删除。