基于非监督特征学习的隧道钢轨表面病害检测方法及装置

摘要

本发明公开了一种基于非监督特征学习的隧道钢轨表面病害检测方法及装置。该方法包括：采集铁路隧道的多张钢轨表面样本图像并扩充至样本数据集；对样本数据集进行预处理，并划分为训练数据集与测试数据集；先提取特征，再进行重构以训练出特征提取器；通过特征提取器建立训练模型；对训练模型进行编译，并将数据投喂至训练网络中进行训练；先在特征提取器中加入训练网络的全连接层，再在全连接层后添加归一化指数函数层，最后将得分最大的样本所对应的类别作为训练网络预测出的类别；验证训练模型，是则模型有效，否则重新对特征提取器进行训练。本发明使检测效率和检测精度大幅提升，减轻人工劳动强度并降低成本，扩充铁路隧道检测数据。