基于改进的RetinaNet网络检测钢轨缺陷的方法和系统

本发明提供了基于改进的RetinaNet网络检测钢轨缺陷的方法，改进EfficientNetB0网络和RetinaNet网络中的主干网络ResNet，形成改进的RetinaNet网络；利用构建的训练集、验证集和测试集对改进的RetinaNet网络进行预训练，得到预训练的改进的RetinaNet网络；采集实际检测区域内的钢轨的图像并将其输入预训练的改进的RetinaNet网络进行缺陷检测和缺陷检测结果输出；将缺陷检测结果与缺陷标准进行比对分析。本发明还提供了一种基于改进的RetinaNet网络检测钢轨缺陷的系统。本发明实现了对钢轨缺陷进行实时检测，降低铁路的维护成本和降低事故率。