一种钢轨表面缺陷检测方法、设备、介质及产品

    公开(公告)号:CN118172312A

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

    申请号:CN202410232576.4

    申请日:2024-02-29

    摘要: 本发明公开一种钢轨表面缺陷检测方法、设备、介质及产品,涉及钢轨表面缺陷检测技术领域,方法包括获取目标钢轨表面图像;将所述目标钢轨表面图像输入至训练后的Densenet卷积神经网络模型,得到所述目标钢轨表面图像中目标钢轨表面缺陷的类别;其中,模型的训练过程为:获取样本钢轨表面图像;对所述样本钢轨表面图像进行目标标记,得到标记后的样本钢轨表面图像;基于所述钢轨表面图像数据集,对改进的Densenet卷积神经网络模型进行训练,得到训练后的模型。本申请通过优化通道融合中的Canny边缘检测方法,对精度进行了提升;同时,采用多尺度特征融合策略,进一步提高了分类的准确性。