一种基于LAH-YOLO的钢轨表面缺陷检测方法

本发明公开了一种基于LAH‑YOLO的钢轨表面缺陷检测方法。YOLOv5‑s算法在钢轨表面缺陷数据集上的检测精度仍需提升。本发明获取钢轨表面图像，将所述钢轨表面图像输入构建好的LAH‑YOLO缺陷检测网络；所述LAH‑YOLO缺陷检测网络包括骨干网络、特征融合网络和检测头；所述骨干网络包括GRF‑C3模块，特征融合网络为AFPN特征融合网络。本发明适用于小目标和小样本数据集，在降低参数量的同时，显著提升了检测精度，在钢轨表面缺陷数据集上具有更好的检测性能。