公开(公告)号：CN119444682A

公开(公告)日：2025-02-14

申请号：CN202411465548.3

申请日：2024-10-21

Applicant: 中铁第一勘察设计院集团有限公司

Inventor： 苏成光 ,  李艳龙 ,  张鹏 ,  张岷 ,  梁栋 ,  褚卫松 ,  卢明 ,  程培涛 ,  方家 ,  姚宇飞 ,  武钰斌 ,  李佳静

IPC: G06T7/00 ,  G06N3/045 ,  G06N3/0464 ,  G06N3/082 ,  G06V10/80 ,  G06V10/82

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