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

权利要求书

1.一种螺栓缺陷数据集扩充方法，其特征在于，所述扩充方法包括：

将数据集中的缺陷螺栓目标裁剪出来并保存；

随机选择裁剪得到的缺陷螺栓图像，并调整图像的尺寸；

确定图像粘贴的中心位置；

计算原始标注框与新标注框的交并比IoU；

IoU的计算公式表示为：

式中，原始标注框为原始巡检图片中原始的标注框，新标注框为在中心位置粘贴图像时的标注框，IoU＝0时进行粘贴；

将裁剪得到的缺陷螺栓图像采用像素填充的方式粘贴到巡检图片中。

2.根据权利要求1所述的一种螺栓缺陷数据集扩充方法，其特征在于，所述确定图像粘贴的中心位置具体包括：

随机采样得到粘贴图像时的中心位置，随机采用的公式表示为：

式中，(x,y)表示随机采样得到的中心位置坐标，crop_w和crop_h表示裁剪图像的宽和高，w和h表示巡检图片的宽和高。

3.根据权利要求1所述的一种螺栓缺陷数据集扩充方法，其特征在于，所述将裁剪得到的缺陷螺栓图像采用像素填充的方式粘贴到巡检图片中具体包括：

像素填充粘贴的公式表示为：

式中，x_A对应原始巡检图片，x_B对应调整尺寸后的裁剪图像，是粘贴后的新图片；M是一个与原始巡检图片维度相同的由0和1组成的矩阵，标记需要粘贴的区域和保留的区域，需要粘贴的区域值均为0，其余位置为1；⊙代表逐像素相乘操作；1是一个与裁剪图像维度相同的元素都为1的矩阵。