[发明专利]一种钢轨损伤检测方法、装置、设备和介质

权利要求书

1.一种钢轨损伤检测方法，其特征在于，包括：

获取待检测图片；

输入所述待检测图片至预先训练得到的钢轨损伤检测模型中，得到损伤类别；

其中，所述钢轨损伤检测模型包括预处理模块和类别检测模块；

所述预处理模块用于对待检测图片进行预处理，得到若干个尺度的池化特征图；

所述类别检测模块用于提取各池化特征图的尺度感知权重并将其映射至所述池化特征图以得到第一融合特征图，提取所述第一融合特征图的空间感知权重并将其映射至所述第一融合特征图中，得到第二融合特征图；识别所述第二融合特征图得到所述损伤类别。

2.根据权利要求1所述的钢轨损伤检测方法，其特征在于，所述对待检测图片进行预处理，得到若干个尺度的池化特征图，进一步包括：

提取所述待检测图片的特征，得到所述待检测图片的初始特征图；

根据预设的尺度对初始特征图进行下采样或上采样，得到多尺度的特征图；

对各尺度特征图进行区域提取及区域池化处理，得到各尺度的区域特征图；

对各尺度的区域特征图进行尺度特征池化处理，得到多个池化特征图。

3.根据权利要求1所述的钢轨损伤检测方法，其特征在于，所述提取各池化特征图的尺度感知权重并将其映射至所述池化特征图以得到第一融合特征图，进一步包括：

将各池化特征图进行平均池化，得到平均池化特征图；

将所述平均池化特征图进行线性卷积处理及非线性变换，得到各池化特征图的尺度感知权重；

将各池化特征图的尺度感知权重与所述池化特征图进行乘积处理，得到所述第一融合特征图。

4.根据权利要求3所述的钢轨损伤检测方法，其特征在于，所述将所述平均池化特征图进行线性卷积处理及非线性变换，得到各池化特征图的尺度感知权重，包括：利用如下公式计算各池化特征图的尺度感知权重：

其中，为个池化特征图的平均值，F为所述池化特征图，S根据所述池化特征图的高度和宽度确定得到，C为所述特征图的通道数，f为线性卷积函数，σ为激活函数。

5.根据权利要求1所述的钢轨损伤检测方法，其特征在于，所述提取所述第一融合特征图的空间感知权重并将其映射至所述第一融合特征图中，得到第二融合特征图，进一步包括：

对所述第一融合特征图分别进行多核线性卷积以及可变形卷积操作，得到空间感知权重；

对所述第一融合特征图和所述空间感知权重进行乘积操作，得到所述第二融合特征图。

6.如权利要求1所述的钢轨损伤检测方法，其特征在于，所述类别检测模块包括多个类别检测支路，每一类别检测支路包括第一轻量级动态检测模块及二分类模块，每一类别检测支路利用一种类别钢轨损伤的历史图片训练得到；

利用第一轻量级动态检测模块提取各池化特征图的尺度特征并将其映射至所述池化特征图以得到第一融合特征图，提取所述第一融合特征图的空间感知权重并将其映射至所述第一融合特征图中，得到第二融合特征图；利用二分类模块识别所述第二融合特征图得到所述损伤类别。

7.如权利要求1所述的钢轨损伤检测方法，其特征在于，所述类别检测模块还包括区域检测支路，所述区域检测支路包括第二轻量级动态检测模块及回归模块，利用所有类别钢轨损伤的历史图片训练得到；

利用第二轻量级动态检测模块提取各池化特征图的尺度特征并将其映射至所述池化特征图以得到第三融合特征图，提取所述第三融合特征图的空间感知权重并将其映射至所述第三融合特征图中，得到第四融合特征图；

利用回归模块识别所述第四融合特征图以确定所述待检测图片中的损伤位置。