一种基于深度学习的铁路坡面排水结构劣化识别方法

本发明提供了一种基于深度学习的铁路坡面排水结构劣化识别方法，包括：通过无人机对坡面排水结构进行俯拍，获得原始排水沟劣化图像；对原始排水沟劣化图像和标注图像集进行数据集增强，建立排水沟劣化图像数据集；构建改进的YOLOv7网络，包括在YOLOv7网络中添加Transformer检测头和注意力机制SE模块；将排水沟劣化图像数据集输入到改进的YOLOv7网络中进行模型训练，利用EIOU函数计算模型损失函数，获得训练后的改进YOLOv7目标检测模型；对待检测的排水沟图像进行排水结构劣化检测，输出包含劣化区域检测框的检测图像。本发明采用无人机拍摄铁路图像，对铁路坡面劣化能够进行快速、有效的识别。