[发明专利]一种螺栓缺陷数据集扩充方法

摘要

本发明提供了一种螺栓缺陷数据集扩充方法，所述扩充方法包括：将数据集中的缺陷螺栓目标裁剪出来并保存；随机选择裁剪得到的缺陷螺栓图像，并调整图像的尺寸；确定图像粘贴的中心位置；计算原始标注框与新标注框的交并比IoU，将裁剪得到的缺陷螺栓图像采用像素填充的方式粘贴到巡检图片中。第一，充分利用已有的螺栓缺陷图片数据，合成了新的数据，增加了训练图片数量，有利于网络模型充分训练，提高其泛化能力；第二，经过数据扩充后，巡检图片中包含有更多的缺陷螺栓目标，不仅增加了缺陷螺栓在图片中的占比，还增加了缺陷螺栓的数量，从而增加了缺陷螺栓目标在计算网络的回归和分类损失时的贡献。