一种螺栓缺陷数据集扩充方法

    公开(公告)号:CN113610838A

    公开(公告)日:2021-11-05

    申请号:CN202110982508.6

    申请日:2021-08-25

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

Patent Agency Ranking