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