[发明专利]一种铁路扣件缺陷检测方法和装置

权利要求书

1.一种铁路扣件缺陷检测方法，其特征在于，包括：

根据采集的铁路扣件图像信息，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本；

根据目标算法对所述第一目标样本进行扩充，确定扩充后的第二目标样本；

根据所述第二目标样本和所述第一目标样本，生成用于检测铁路扣件的检测模型；

将所述待检测样本输入所述检测模型，确定检测结果。

2.根据权利要求1所述的方法，其特征在于，根据采集的铁路扣件图像信息，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本，包括：

获取铁路扣件图像信息，所述铁路扣件图像信息包括深度图像和强度图像；所述深度图像和强度图像的像素一一对应；

根据所述深度图像的像素，确定所述铁路扣件图像信息中铁路扣件的位置信息；

根据所述铁路扣件的位置信息和所述强度图像的像素，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本。

3.根据权利要求2所述的方法，其特征在于，根据所述深度图像的像素，确定所述铁路扣件图像信息中铁路扣件的位置信息，包括：

根据所述深度图像的像素，确定铁路扣件区域和铁轨区域；

根据所述铁路扣件区域和铁轨区域，确定所述铁路扣件图像信息中铁路扣件的位置信息。

4.根据权利要求3所述的方法，其特征在于，所述根据所述铁路扣件区域和铁轨区域，确定所述铁路扣件图像信息中铁路扣件的位置信息，包括：

根据所述铁路扣件区域和所述铁轨区域，确定所述铁路扣件的横向坐标；

根据所述横向坐标和所述深度图像，确定所述铁路扣件的横向位置；

其中，所述横向坐标满足以下公式：

x_k∈[x_a-d_x,x_a]∪[x_a+h,x_a+h+d_x]；其中，x_k表示所述铁路扣件区域的横向坐标，x_a为所述铁轨区域的铁轨一侧面的横向坐标，h为所述铁轨区域的铁轨宽度，d_x为第一预设阈值。

5.根据权利要求4所述的方法，其特征在于，所述根据所述铁路扣件区域和铁轨区域，确定所述铁路扣件图像信息中铁路扣件的位置信息，包括：

根据所述深度图像，将确定的所述铁路扣件的横向位置，沿所述铁路扣件的横向方向滤波，确定所述铁路扣件的目标高度阈值和目标高度曲线；

根据所述目标高度阈值和目标高度曲线，确定所述铁路扣件的纵向坐标；

其中，所述纵向坐标满足以下公式：

y_k∈[y₁-Δy,y₂+Δy]；其中，y_k表示所述铁路扣件区域的纵向坐标，y₁和y₂表示所述铁轨区域的铁轨的高度坐标，△y为第二预设阈值。

6.根据权利要求2所述的方法，其特征在于，根据所述铁路扣件的位置信息和所述强度图像的像素，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本，包括：

根据所述强度图像的像素，对所述铁路扣件的区域进行均值滤波；

根据滤波后的所述铁路扣件和所述铁路扣件的位置信息，确定具有位置信息的待检测样本以及具有位置信息的第一目标样本。

7.根据权利要求1所述的方法，其特征在于，所述根据目标算法对所述第一目标样本进行扩充，确定扩充后的第二目标样本，包括：

根据生成式对抗神经网络算法对所述第一目标样本的缺陷扣件样本进行扩充，确定扩充后的第二目标样本；

其中，所述生成式对抗神经网络算法包括生成器和判别器；所述生成器用于根据缺陷扣件样本进行扩充；所述判别器用于确定扩充样本和原始样本的类别。