[发明专利]一种铁轨扣件状态检测方法及系统

权利要求书

1.一种铁轨扣件状态检测方法，其特征在于，包括：

获取待识别铁轨扣件图像；

将所述待识别铁轨扣件图像输入至第一铁轨环境分割模型以及第二铁轨环境分割模型；

所述第一铁轨环境分割模型响应于输入的所述待识别铁轨扣件图像，输出第一坐标信息，所述第一坐标信息包括螺母的中心点坐标信息以及轮廓点坐标信息、铁轨的中心点坐标信息以及轮廓点坐标信息；

所述第二铁轨环境分割模型响应于输入的所述待识别铁轨扣件图像，输出第二坐标信息，所述第二坐标信息包括待识别铁轨扣件的中心点坐标信息以及轮廓点坐标信息；

根据所述第二坐标信息，得到待识别铁轨扣件局部图像；

将所述待识别铁轨扣件局部图像输入至扣件状态检测模型；

所述扣件状态检测模型响应于输入的所述待识别铁轨扣件局部图像，输出待识别铁轨扣件图像的初始预测状态；

根据所述第一坐标信息和所述第二坐标信息，得到待识别铁轨扣件图像的状态辅助判断信息，其中所述状态辅助判断信息为缺损或者倾斜或者正常；

当所述状态辅助判断信息为缺损时，输出所述待识别铁轨扣件图像的状态类别为缺损；

当所述状态辅助判断信息为倾斜时，输出所述待识别铁轨扣件图像的状态类别为倾斜；

当所述状态辅助判断信息为正常时，输出所述待识别铁轨扣件图像的状态类别为所述初始预测状态；

其中，根据所述第一坐标信息和所述第二坐标信息，得到待识别铁轨扣件图像的状态辅助判断信息，包括：

根据所述第一坐标信息和所述第二坐标信息，得到第一面积、第二面积、第三面积和第四面积，其中，所述第一面积为以螺母为中心划分的扣件的一侧区域与铁轨交叠的面积，所述第二面积为以螺母为中心划分的扣件的另一侧区域与铁轨交叠的面积，所述第三面积为以螺母为中心划分的扣件一侧区域的面积，所述第四面积为以螺母为中心划分的扣件另一侧区域的面积；

当第一比值大于第一预设比值时，所述状态辅助判断信息为倾斜，其中所述第一比值为所述第一面积与所述第二面积的差值和所述第一面积与所述第二面积的和值的比值；

当第二比值大于第二预设比值时，所述状态辅助判断信息为缺损，其中所述第二比值为所述第三面积与所述第四面积的差值和所述第三面积与第四面积的和值的比值；

当所述第一坐标信息中的螺母中心点坐标信息为空且所述第二坐标信息中的扣件中心点坐标信息不为空时，所述状态辅助判断信息为缺损。

2.根据权利要求1所述的铁轨扣件状态检测方法，其特征在于，所述第一铁轨环境分割模型按照以下方法训练：

获取多个铁轨图像；

采用像素级标注的方式标注所述铁轨图像中的螺母轮廓以及铁轨轮廓，得到第一标注后的铁轨图像；

对多个所述第一标注后的铁轨图像执行预处理过程，得到第一训练数据集；

使用所述第一训练数据集，按照基于极坐标系的实例分割算法训练第一环境分割模型；

所述第二铁轨环境分割模型按照以下方法训练；

获取多个铁轨图像；

采用像素级标注的方式标注所述铁轨图像中的扣件轮廓，得到第二标注后的铁轨图像；

对多个所述第二标注后的铁轨图像执行预处理过程，得到第二训练数据集；

使用所述第二训练数据集，按照基于极坐标系的实例分割算法训练第二环境分割模型。

3.根据权利要求2所述的铁轨扣件状态检测方法，其特征在于，所述第一铁轨环境分割模型得到的螺母的轮廓点的数量为48个，铁轨的轮廓点的数量为48个；

所述第二铁轨环境分割模型得到扣件的轮廓点的数量为48个。

4.根据权利要求1所述的铁轨扣件状态检测方法，其特征在于，还包括：将所述待识别铁轨扣件图像以及对应的状态类别发送至服务器。