钢轨缺陷检测模型的训练方法及钢轨缺陷的检测方法

本申请提供了一种钢轨缺陷检测模型的训练方法及钢轨缺陷的检测方法，方法包括：获取包括钢轨的训练数据，并对训练数据中的每个图像进行标注，得到图像的标注结果，通过特征提取层，提取每个图像的第一特征图，基于跨尺度融合机制中的卷积操作，对第一特征图的通道数进行调整，得到第二特征图，基于重参数化卷积操作，并联提取第二特征图中不同尺度的尺度信息，并将每个尺度对应的尺度信息进行融合，得到与第二特征图对应的第三特征图，将第二特征图和第三特征图进行元素级相加操作，得到融合特征图，并基于融合特征图、图像的标注结果和预设的损失函数对钢轨缺陷检测模型进行训练。