首页|期刊导航|中国铁道科学|基于ADF-YOLO模型的双模态轨道扣件缺陷检测方法

基于ADF-YOLO模型的双模态轨道扣件缺陷检测方法OA

Dual-modal rail fastener defect detection model based on ADF-YOLO

中文摘要英文摘要

针对轨道扣件缺陷检测中单一可见光图像易受环境干扰、缺乏空间结构信息,导致模型对小尺寸缺陷及扣件空间形变敏感性不足,进而易产生误检和漏检的问题,提出一种基于注意力机制与双模态融合YOLO(Attention and Dual-Modal Fusion YOLO,ADF-YOLO)模型的检测方法.首先,构建ADF-YOLO模型,通过双流骨干网络结构,并行提取轨道扣件可见光图像的纹理特征和深度图像的空间结构特征;其次,为提高对小尺寸缺陷特征的捕获能力,在ADF-YOLO模型骨干网络中引入注意力与多分支卷积协同模块,扩大感受野,强化细节特征表达;最后,为实现对异构特征的高效融合,在ADF-YOLO模型颈部网络引入注意力引导的双模态特征融合模块,充分挖掘双模态特征融合范式,生成鲁棒性更高且对缺陷形态更敏感的融合特征.结果表明:在扣件缺陷多模态数据集上,基于ADF-YOLO模型的检测方法,在交并比阈值为0.5和0.5~0.95时平均精度均值分别达90.9%和70.5%;与多种单模态和多模态目标检测模型相比,基于ADF-YOLO模型的检测方法在检测精度上具有显著优势.提出的ADF-YOLO模型可为轨道扣件缺陷巡检任务提供可靠的技术支撑.

In rail fastener defect detection,single-modal visible-light images are susceptible to environmental interference and lack spatial structural information.This limitation reduces the model's sensitivity to small-sized defects and spatial deformations of fasteners.Consequently,it increases the rate of false and missed detections.To address these issues,this paper proposes a detection method based on the Attention and Dual-Modal Fusion YOLO(ADF-YOLO)model.First,the ADF-YOLO model is constructed,which employs a dual-stream backbone architecture to concurrently extract texture features from visible-light images and spatial structural features from depth images of railway fasteners.Second,an Attention and Multi-Branch Convolution Synergy Module is introduced into the backbone network to improve the capture of small-sized defect features.This module expands the receptive field and enhances the representation of detailed features.Finally,an Attention-Guided Dual-Modal Feature Fusion Module is integrated into the neck network for the efficient fusion of heterogeneous features.This module fully exploits the dual-modal feature fusion paradigm.Consequently,it generates more robust fused features that are highly sensitive to defect morphology.The results demonstrate that on a multimodal fastener defect dataset,the ADF-YOLO model detection method achieves a mean average precision(mAP)of 90.9%and 70.5%at IoU thresholds of 0.5 and 0.5-0.95,respectively.The ADF-YOLO method significantly outperforms various single-modal and multi-modal object detection models in detection accuracy.The proposed ADF-YOLO model provides reliable technical support for railway fastener defect inspection tasks.

秦勇;彭根旺;曹志威;高阳;寇淋淋;王雅观

北京交通大学 先进轨道交通自主运行全国重点实验室,北京 100044北京交通大学 先进轨道交通自主运行全国重点实验室,北京 100044||北京交通大学 交通运输学院,北京 100044北京交通大学 先进轨道交通自主运行全国重点实验室,北京 100044北京交通大学 先进轨道交通自主运行全国重点实验室,北京 100044||北京交通大学 交通运输学院,北京 100044北京地铁运营有限公司,北京 100044北京地铁运营有限公司技术创新研究院分公司,北京 100044

交通工程

铁路安全轨道扣件缺陷检测多模态融合注意力机制YOLO

Railway safetyRail fastenersDefect detectionMultimodal fusionAttention mechanismYOLO

《中国铁道科学》 2026 (4)

105-118,14

先进轨道交通自主运行全国重点实验室自主课题(RAO2025ZR001)北京市自然科学基金资助项目(L241078)国家资助博士后研究人员计划项目(GZC20251118)

10.3969/j.issn.1001-4632.2026.04.08

评论