一种铁路轨枕裂纹目标检测模型训练方法、装置及设备

    公开(公告)号:CN118447349A

    公开(公告)日:2024-08-06

    申请号:CN202410621447.4

    申请日:2024-05-20

    Abstract: 本发明适用于轨枕裂纹目标检测技术领域,提供了一种铁路轨枕裂纹目标检测模型训练方法、装置及设备,所述方法包括:获取轨枕裂纹目标图像,并对轨枕裂纹目标图像进行标注,形成训练数据集;将训练数据集当中的训练图像输入到预设的轨枕裂纹目标检测网络当中进行训练;根据预测的轨枕裂纹目标检测信息与标注的真实轨枕裂纹目标检测信息,计算模型损失,并根据模型损失且采用反向传播算法迭代更新所述轨枕裂纹目标检测网络参数。本发明通过对轨枕裂纹目标检测网络进行优化,使得在特征提取之后,会对特征依序进行NAS最优特征融合增加网络对图像特征的表达能力,同时利用TOOD Head模型增加对轨枕裂纹特征分类与回归之间的交互,进一步提升NAS‑Swin检测性能,能够解决小目标轨枕裂纹的检测存在精度、召回率、速度较低等问题,提高模型检测精度和泛化能力。

Patent Agency Ranking