浙江大学学报(工学版), 2026, 60(11): 2453-2461 doi: 10.3785/j.issn.1008-973X.2026.11.014

计算机技术

面向小目标与模糊目标的轨道缺陷检测框架

黄腾飞,, 火久元,, 曹芳, 武泽宇, 孔垂乐

1. 兰州交通大学 电子与信息工程学院,甘肃 兰州 730070

2. 国家冰川冻土沙漠科学数据中心,甘肃 兰州 730000

Framework for railway defect detection oriented to small and blurry object

HUANG Tengfei,, HUO Jiuyuan,, CAO Fang, WU Zeyu, KONG Chuile

1. School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China

2. National Cryosphere Desert Data Center, Lanzhou 730000, China

通讯作者: 火久元,男,教授,博导. orcid.org/0000-0003-2395-4133. E-mail:huojy@mail.lzjtu.cn

收稿日期: 2025-09-26  

基金资助: 国家自然科学基金资助项目(62262038);甘肃省科技计划-中小企业创新基金资助项目(25CXGA014);甘肃省科技创新引导计划-科技专员专项(25CXGA030).

Received: 2025-09-26  

Fund supported: 国家自然科学基金资助项目(62262038);甘肃省科技计划-中小企业创新基金资助项目(25CXGA014);甘肃省科技创新引导计划-科技专员专项(25CXGA030).

作者简介 About authors

黄腾飞(1996—),男,博士生,从事轨道缺陷检测研究.orcid.org/0009-0005-3588-6803.E-mail:849138454@qq.com , E-mail:849138454@qq.com

摘要

为了解决轨道缺陷检测中小目标易漏检与模糊目标易误检的问题,提出动态多尺度与纹理感知框架(DMT). 从动态尺度感知池化(DSP)、纹理对比增强(TCE)和模糊边缘感知(BBA)出发,系统提升模型在复杂场景下的特征提取与边缘表征能力. 基于DMT框架,设计DSP、TCE与BBA 3种轻量级子模块融入检测网络. 将所提框架嵌入YOLOv8,构建名为DMT-YOLOv8的轨道缺陷检测模型. 在真实轨道缺陷数据集上的实验结果表明,DMT-YOLOv8在小目标检测与模糊目标检测方面均优于基线模型,具备良好的通用性与泛化性. DMT-YOLOv8的mAP@0.5与mAP@0.5:0.95分别达到89.1%和50.2%,整体性能超越多种主流检测方法.

关键词: 轨道缺陷检测 ; 小目标检测 ; 模糊目标 ; 多尺度特征融合 ; 纹理对比增强 ; 边缘模糊感知

Abstract

To address the problems of missed detection of small objects and false detection of blurred objects in railway defect detection, a dynamic multi-scale and texture-aware framework (DMT) was proposed. Through dynamic scale-aware pooling (DSP), texture contrast enhancement (TCE), and blurred boundary awareness (BBA), the model’s feature extraction and boundary representation capabilities were enhanced in complex scenarios. Based on the DMT framework, three lightweight modules, DSP, TCE, and BBA, were designed and integrated into the detection network. The proposed framework was incorporated into YOLOv8 to construct a railway defect detection model termed DMT-YOLOv8. Experimental results on real-world railway defect datasets showed that DMT-YOLOv8 outperformed the baseline model in both small-object and blurred-object detection and exhibited good generality and generalization capability. Specifically, the mAP@0.5 and mAP@0.5:0.95 reached 89.1% and 50.2%, respectively, and the overall performance surpassed that of several mainstream detection methods.

Keywords: railway defect detection ; small object detection ; blurry object ; multi-scale feature fusion ; texture contrast enhancement ; blurred-boundary awareness

PDF (6359KB) 元数据 多维度评价 相关文章 导出 EndNote| Ris| Bibtex  收藏本文

本文引用格式

黄腾飞, 火久元, 曹芳, 武泽宇, 孔垂乐. 面向小目标与模糊目标的轨道缺陷检测框架. 浙江大学学报(工学版)[J], 2026, 60(11): 2453-2461 doi:10.3785/j.issn.1008-973X.2026.11.014

HUANG Tengfei, HUO Jiuyuan, CAO Fang, WU Zeyu, KONG Chuile. Framework for railway defect detection oriented to small and blurry object. Journal of Zhejiang University(Engineering Science)[J], 2026, 60(11): 2453-2461 doi:10.3785/j.issn.1008-973X.2026.11.014

长期以来,铁路轨道的运行安全问题始终是交通运输领域的关注重点. 随着铁路运输系统的发展,钢轨在长期服役过程中,不可避免地会出现裂缝[1]、磨损和腐蚀等缺陷,严重威胁列车运行的稳定性与行车安全[2]. 轨道线路具有里程长和分布广的特点,传统依赖人工巡检的方式不仅效率低、成本高,且存在较强的主观性与安全隐患[3]. 为了提高检测效率与准确性,研究者逐渐将自动化与智能化技术引入轨道巡检系统[4].

早期的轨道检测技术主要依赖物理方法(如复合电磁[5]和震动信号[6]),这些方法在特定环境下具有一定的效果,但受制于固定设备和适用范围,难以进行大规模推广. 随着计算机视觉与图像处理技术的发展,出现了基于图像的轨道缺陷检测方法. 最初的图像检测方法多采用小波变换、边缘检测和阈值分割等手工特征提取方式[7],这些方法缺乏对图像高级语义的理解,面对复杂环境干扰或模糊目标时,往往检测效果不佳[8]. 深度学习技术的迅速发展为轨道缺陷检测提供了新的解决思路. 基于卷积神经网络(CNN)的目标检测方法[9-11],通过端到端的方法来学习图像中的高级语义特征,在多目标检测任务中取得了优异的性能. 然而,这类算法在应用到轨道缺陷检测领域时,仍面临实际困难. 一方面,轨道缺陷种类多且尺寸差异大,较小缺陷(如裂缝和腐蚀)易被漏检[12-15];另一方面,磨损和挤压缺陷往往边界模糊、纹理变化不明显且对比度低,易被误判为背景信息[8,16-17]. 针对上述问题,部分研究尝试引入注意力机制[18-20]与多尺度结构[21-24]来增强模型的感知能力,但存在以下不足:1)在现有的特征提取模块中,较小的缺陷所占像素较少,易在网络下采样阶段丢失或弱化[25],后期被漏检;2)边缘模糊类缺陷的识别效果依赖于缺陷的边缘清晰度,当前的研究缺乏处理弱边缘与弱纹理的方法[26],造成缺陷的错检和漏检.

本研究提出动态多尺度与纹理感知框架(dynamic multi-scale and texture-aware,DMT),为复杂缺陷场景下的检测提供系统化解决方案;为了提升网络对小目标与模糊缺陷的表征能力,基于DMT框架设计3种轻量级模块,分别为动态尺度感知池化(dynamic scale-aware pooling, DSP)、纹理对比增强(texture contrast enhancement, TCE)与模糊边缘感知(blurred boundary awareness, BBA);将DMT嵌入YOLOv8,构建轨道缺陷检测模型DMT-YOLOv8,在真实轨道缺陷数据集上开展实验,验证所提框架及模块的有效性.

1. 方法设计

1.1. 动态多尺度与纹理感知框架概述

DMT旨在从特征建模层面对检测网络进行有针对性的增强,通过多维度感知优化来提升对小目标与模糊目标的识别能力. DMT包含3个核心维度. 1)动态尺度建模(DSP):针对小目标在下采样过程中易出现特征稀释与语义丢失的问题,自适应调整感受野以增强其在深层特征中的表达. 2)纹理对比增强(TCE):面向边界模糊(如磨损和腐蚀)或低对比区域,突出纹理差异,避免其被误判为背景. 3)边界感知(BBA):对于模糊边缘类缺陷,强化网络对空间结构的建模能力,并与纹理增强形成互补. 这3个维度协同优化的DMT框架能够为轨道缺陷检测中小目标与模糊目标检测提供系统化的解决方案.

1.2. 动态尺度感知池化模块

在典型的目标检测网络中,特征提取模块通过多层下采样与语义抽象构建特征,小目标所占像素较少,易在网络下采样阶段丢失或弱化,后期被漏检. 此外,常规卷积操作在所有通道间共享感受野,缺乏结构差异性建模能力,进一步削弱了小目标的表达能力.

DSP能够增强模型对小目标缺陷的感知能力:引入通道分组机制,实现通道关注的差异化建模,减少通道间特征干扰;引入多尺度空洞卷积机制,强化小目标特征表征. DSP结构如图1所示,将输入特征图表示为$ \boldsymbol{X}\in {{{\mathbf{R}}}}^{C\times H\times W} $,其中$ C$、$ H$、$ W$分别为通道数、高度和宽度,则有分组策略表达式为

图 1

图 1   动态尺度感知池化模块的网络结构

Fig.1   Network structure of dynamic scale-aware pooling module


$ \boldsymbol{X}\in {{{\mathbf{R}}}}^{C\times H\times W}\rightarrow {\boldsymbol{X}}_{n}\in {\mathbf{R}}^{\tfrac{C}{N}\times H\times W} . $

式中:$ N$为预设的分组大小;$ {\boldsymbol{X}}_{n} $为输入特征的子集,$ {\boldsymbol{X}}_{n}\in \left[{\boldsymbol{X}}_{1}\mathbf{,}{\boldsymbol{X}}_{2}\mathbf{,}\cdots \mathbf{,}{\boldsymbol{X}}_{N}\right] $. 对各子集中的特征图施加不同膨胀率的空洞卷积操作:

$\begin{split} {\boldsymbol{Y}}_{n}\left(c,i,j\right)=\sum \limits_{u=1}^{k}\sum \limits_{v=1}^{k}{\boldsymbol{W}}_{n}\left(c,u,v\right){\boldsymbol{X}}_{n}\left(c,i+{r}_{n}u,j+{r}_{n}v\right)\end{split} . $

式中:$ {\boldsymbol{Y}}_{n} $为输出特征图,$ c$为通道索引,$ (i,\;j)$为空间通道,$ k$为卷积核大小,$ {\boldsymbol{W}}_{n} $为该组对应的卷积核权重,$ r_n$为该组使用的膨胀率. 在完成分组动态空洞卷积后,各组输出$ {\boldsymbol{Y}}_{n} $沿通道维度进行拼接,形成融合输入;通过1×1卷积操作实现通道压缩与信息聚合,输出融合特征图$ \boldsymbol{Z} $:

$ \boldsymbol{Y}=\text{Concat}\left({\boldsymbol{Y}}_{1}\mathbf{,}{\boldsymbol{Y}}_{2}\mathbf{,}\cdots \mathbf{,}{\boldsymbol{Y}}_{n}\mathbf{,}\cdots \mathbf{,}{\boldsymbol{Y}}_{N}\right) , $

$ \boldsymbol{Z}={\text{Conv}}_{1\times 1}\left(\boldsymbol{Y}\right) . $

引入如图2所示的轻量级通道注意力机制(efficient channel attention,ECA)来增强关键通道的响应能力. 在完成ECA增强后,输出与原始输入进行残差连接,残差结果依次通过BN层和ReLU激活函数处理,最终生成模块的输出特征图. ECA及后续操作表达式为

图 2

图 2   轻量级通道注意力机制的网络结构

Fig.2   Network structure of efficient channel attention mechanism


$ \boldsymbol{Q}=\mathrm{Re}\text{LU}\left(\text{BN}\left(\text{ECA}\left(\boldsymbol{Z}\right)+\boldsymbol{X}\right)\right) . $

其中+表示残差连接. 通过引入通道分组和多尺度卷积,DSP在保持计算开销可控的前提下,能够有效缓解小目标特征稀疏与竞争干扰问题,显著提升特征表征质量.

1.3. 纹理对比增强模块

在铁路轨道缺陷检测任务中,如磨损和腐蚀的缺陷通常具有边界模糊和纹理变化微弱的特征. 常规检测网络难以准确建模此类区域中的局部突变,易被误判为背景信息,导致漏检. 如图3所示,TCE采用双分支结构,并行构建局部分支与全局分支,分别捕捉细节突变与上下文信息;通过差异建模生成纹理对比图,提取显著纹理特征,以引导主干网络更准确地聚焦于模糊区域中的潜在缺陷. 构造并行的局部与全局纹理提取分支,采用不同尺度的卷积核(3×3、5×5)对输入特征图$ \boldsymbol{X}\in {{\mathbf{R}}}^{C\times H\times W} $进行卷积操作,分别提取局部纹理响应特征$ {\boldsymbol{T}}_{\text{L}} $与全局纹理相应特征$ {\boldsymbol{T}}_{\text{G}} $,构造两者之间的纹理差异图$ \boldsymbol{D} $,以刻画不同感受野下的响应不一致区域,突出潜在模糊目标的特征突变:

图 3

图 3   纹理对比增强模块的网络结构

Fig.3   Network structure of texture contrast enhancement module


$ {\boldsymbol{T}}_{\text{L}}={\text{Conv}}_{3\times 3}\left(\boldsymbol{X}\right) , $

$ {\boldsymbol{T}}_{\text{G}}={\text{Conv}}_{5\times 5}\left(\boldsymbol{X}\right) , $

$ \boldsymbol{D}=\left| {\boldsymbol{T}}_{\text{L}}-{\boldsymbol{T}}_{\text{G}}\right| . $

在差异建模后引入由1×1卷积、BN层以及ReLU激活函数组成的编码结构. 该结构可有效压缩通道冗余,突出目标边缘区域:

$ \boldsymbol{E}=\mathrm{Re}\text{LU}\left(\text{BN}\left({\text{Conv}}_{1\times 1}\left(\boldsymbol{D}\right)\right)\right) . $

在此基础上,融合ECA模块,进一步构建纹理引导权重图,将权重图与原始输入特征进行逐元素乘法融合,实现对模糊区域的显著性增强:

$ \boldsymbol{Q}=\boldsymbol{X}\odot \text{ECA}\left(\boldsymbol{E}\right) . $

式中:$ \odot$为逐元素相乘. TCE通过局部–全局对比建模纹理差异,有效增强网络在模糊区域的表征能力;该模块结构简洁且计算开销低,具备良好的通用性与泛化能力.

1.4. 模糊边缘感知模块

如图4所示,BBA通过引入弱边缘引导机制与边缘注意力建模,增强特征图中模糊边界区域的边界特征,提升模型在复杂背景下的边界判别能力. 与侧重纹理感知的TCE不同,BBA更注重边界轮廓信息,二者在功能上形成互补. BBA由2个互补分支组成:边缘感知分支和空间注意力分支. 前者用于捕捉模糊区域的边界信息,后者通过注意力机制突出这些区域. 1)边缘感知分支对输入Xc沿通道维度进行压缩,获得单通道特征均值图:

图 4

图 4   模糊边缘感知模块的网络结构

Fig.4   Network structure of blurred boundary awareness module


$ {\boldsymbol{G}}_{\text{Avg}}=\dfrac{1}{C}\sum \limits_{c=1}^{C}{\boldsymbol{X}}_{c} . $

对均值图GAvg施加Sobel算子以计算图像梯度变化,提取边缘变化区域,得到候选的边缘响应图. 具体来说,在BBA中,采用水平与垂直双方向的Sobel算子,以覆盖轨道缺陷的不同走向. 算子采用固定的3×3卷积核,分别计算x与y方向的梯度响应,通过梯度平方和开方得到最终的边缘响应图:

$ {\boldsymbol{E}}_{\text{sobel}}=\text{Sobel}\left({\boldsymbol{G}}_{\text{Avg}}\right) . $

该响应图可用于表征模糊缺陷可能位置区域,作为边缘引导先验. 2)空间注意力分支对输入特征图Xc在通道维度上施加平均池化与最大池化操作,得到2幅单通道特征图GAvg与GMax,在进行拼接后送入7×7卷积进行特征整合,使用BN层和Sigmoid激活函数生成空间注意力图Aspatial:

$ {\boldsymbol{G}}_{\text{Max}}=\max \left({\boldsymbol{X}}_{c}\right) , $

$ {\boldsymbol{A}}_{\text{conv}}={\text{Conv}}_{7\times 7}\left(\text{Concat}\left({\boldsymbol{G}}_{\text{Avg}}\;|\;{\boldsymbol{G}}_{\text{Max}}\right)\right) , $

$ {\boldsymbol{A}}_{\text{spatial}}=\text{Sigmoid}\left(\text{BN}\left({\boldsymbol{A}}_{\text{conv}}\right)\right) . $

将边缘响应图与空间注意力图进行逐元素相乘融合,构建最终的边缘引导权重图Aedge,并用于对原始特征图进行加权增强,实现边缘区域的显著性增强:

$ {\boldsymbol{A}}_{\text{edge}}={\boldsymbol{E}}_{\text{sobel}}\odot {\boldsymbol{A}}_{\text{spatial}} , $

$ \boldsymbol{Q}=\boldsymbol{X}\odot {\boldsymbol{A}}_{\text{edge}} . $

BBA融合边缘引导先验与空间注意力,构建出具有结构感知能力的边缘引导权重图,能够有效提升对于模糊边界的检测性能.

1.5. DMT-YOLOv8轨道缺陷检测模型

为了验证所提模块的有效性,选取YOLOv8作为基准算法,将DSP、TCE与BBA这3个特征增强模块嵌入YOLOv8主干网络中,构造轨道缺陷检测模型DMT-YOLOv8. 经过多轮对比实验,最终确定的最优集成方式如图5所示. 所设计模块被嵌入至主干网络的不同层级:DSP位于P1与P2之间,TCE位于P2与P3之间,BBA位于P4与P5之间. 在整体框架中,DSP、TCE与BBA按照由浅至深的层级顺序依次发挥作用:DSP通过早期增强保留小目标的细粒度特征,为后续模块提供更完整的基础信息;TCE在中层阶段基于DSP传递的特征进行局部与全局纹理差异建模,为后续边界感知提供纹理线索;BBA位于中深层特征图中,在融合来自前层的纹理与空间信息后,进一步强化边界结构的表达. 三者在信息流上形成由细节到语义的递进关系,使特征增强、纹理建模与边界感知过程实现协同优化.

图 5

图 5   DMT-YOLOv8轨道缺陷检测模型的架构

Fig.5   Architecture of DMT-YOLOv8 model for railway defect detection


2. 实验结果及分析

实验均基于Windows 11操作系统与PyTorch框架实现,在搭载Intel i9-14900HX处理器NVIDIA RTX 4060 Laptop GPU的笔记本环境中完成.

2.1. 数据集

采用来自RoboFlow平台的开源数据集Railway Dataset[27]作为实验数据,数据集共包含图像829张,涵盖8类常见轨道缺陷,分别为crush(压痕)、dent(凹陷)、scratch(划痕)、slant(倾斜)、damage(破损)、dirt(污渍)、gap(间隙)和unknown(未知). 为了保证评估稳健性,将数据集按比例8∶2分为训练集(663张)和验证集(166张),在模型训练过程中,仅使用YOLO系列算法内置的包括随机缩放、翻转和颜色扰动在内的多种数据增强策略.

2.2. 评估指标

选取精确率P、召回率R、F1值和2种平均精度均值(mAP@0.5与mAP@0.5:0.95)作为主要评价指标,考虑到轨道缺陷多为中小尺度目标,且大部分缺陷本身具有模糊特征,不再额外引入独立的小目标或模糊目标评价指标.

2.3. 消融实验

为了确定DSP中分组数量N对算法性能的影响,分别选取N=[2,4,8]进行对比实验,结果如表1所示. 可以看出,当N=4时,模型在P、R、F1及mAP上均取得最佳表现,说明此时特征分组与组间信息交互之间达到较优平衡. 当N=2时,特征提取能力不足,导致检测精度与召回率下降;当N=8时,组间特征交互受限,整体性能略有回落. 综上,后续实验统一采用N=4作为默认配置.

表 1   动态尺度感知池化不同分组对所提模型检测性能的影响

Tab.1  Effect of different groups in dynamic scale-aware pooling on detection performance of proposed model

NP/%R/%F1/%mAP@0.5/%mAP@0.5:0.95/%
285.577.981.586.346.5
491.181.285.989.150.2
887.578.582.787.147.5

新窗口打开| 下载CSV


为了确定ECA中不同卷积核大小K对算法性能的影响,分别选取K=[3,5,7]进行对比实验,结果如表2所示. 由实验结果可见,当K=3时,模型在P、R、F1及mAP上均取得最佳表现,说明较小的局部卷积核能够更有效地建模通道间的相关性. 当K进一步增大时,注意力范围过宽,导致通道响应被平滑化,模型对细节特征的敏感性下降,从而引起性能退化. 综上,后续实验统一采用K=3作为ECA模块的默认配置.

表 2   轻量级通道注意力机制不同卷积核大小对所提模型检测性能的影响

Tab.2  Effect of different convolution kernel sizes in efficient channel attention mechanism on detection performance of proposed model

KP/%R/%F1/%mAP@0.5/%mAP@0.5:0.95/%
391.181.285.989.150.2
585.377.181.086.145.9
783.273.878.281.741.9

新窗口打开| 下载CSV


为了验证所提3个模块的独立贡献与协同效果,在YOLOv8基线模型上逐一嵌入DSP、TCE、BBA,并测试不同组合方式对检测性能的影响,结果如表3所示,其中np为参数量,rf为帧率. 考虑到模糊性广泛存在于轨道缺陷领域中,且中小型缺陷为主要缺陷类型,因此不再单独设立小目标与模糊目标指标. 可以看出,所设计模块在单独使用和组合使用的情况下,均能有效提升模型在轨道缺陷检测任务中的整体性能. 模型整体计算开销保持稳定,充分体现了所设计模块的轻量化特性. 在单模块实验中,DSP能够有效提升模型对小目标的检测能力,使得召回率相较基准模型提升了4.9个百分点,展现了该模块在减少漏检方面的有效性;TCE通过局部与全局纹理信息的差异建模,精确率和召回率均实现稳定提升,说明该模块具有较强的普适增强能力;BBA对于精确率贡献最大,相比基准算法提升了2.2个百分点,这说明对边界信息的增强能有效降低模糊目标的误检率. 实验结果与模块的设计初衷相一致:DSP面向小目标表征增强,TCE提供普适纹理建模,BBA强调边界精细化,实验结果印证了各自的针对性作用. 在多模块组合实验中,任意2种模块的组合均能获得相较单模块更优的性能,验证了3种模块之间的互补性. 例如,“DSP+TCE”与“DSP+BBA”的组合均能使mAP@0.5:0.95获得较大提升,说明DSP的小目标增强作用与TCE的差异建模和BBA的边缘引导均能够形成良好互补. 当3种模块同时集成时,模型性能达到最优:mAP@0.5由85.1%提升至89.1%,mAP@0.5:0.95由46.1%提升至50.2%,充分验证本研究提出的多模块协同增强策略在提升轨道缺陷检测精度方面的有效性与实用价值.

表 3   不同模块组合对模型检测性能的影响

Tab.3  Effect of different module combinations on model detection performance

序号DSPTCEBBAP/%R/%F1/%mAP@0.5/%mAP@0.5:0.95/%ΔmAP@0.5:0.95/%np/106rf/(帧·s−1)
1———85.672.378.485.146.1—3.1677.5
2√——86.977.281.886.648.4+2.23.1677.5
3—√—87.076.581.485.447.8+1.63.1976.3
4——√87.874.280.486.447.5+1.33.1677.4
5√√—88.080.984.388.449.2+3.03.1976.3
6√—√88.279.283.587.949.1+2.93.1677.4
7—√√88.778.783.487.848.4+2.23.1976.2
8√√√91.181.285.989.150.2+4.13.1976.2

新窗口打开| 下载CSV


2.4. 模块通用性与对比实验

为了进一步验证所提框架的通用性与可迁移性,额外选取具有代表性的两阶段检测算法Faster R-CNN(ResNet50+FPN)与单阶段检测算法YOLOv5n作为基准模型,将DMT嵌入其中进行实验,评估该框架在不同检测框架下的轻量级特性与性能表现,实验结果如表4所示. 可以看出,所提框架在不同检测模型中均展现出良好的轻量级特性与较强的性能增益. 对于YOLOv8和YOLOv5n,DMT能够在参数量仅增加1%~2%和计算速度损失不超过6%的情况下明显提升精度. 其中精确率、召回率和F1值均有提升,在YOLOv8中,召回率提升近9个百分点,展现了DMT在减少小目标漏检方面的有效性;同时,mAP也增加了3~4个百分点,证实了该框架在单阶段算法中具有良好的适用性. 对于Faster R-CNN (ResNet50+FPN),由于传统两阶段检测器在复杂场景下特征表达能力不足,其基准算法精度明显低于YOLO系列算法,在引入DMT框架后,其精确率、召回率和F1值均有超过10个百分点的提升,mAP@0.5增加了16%,mAP@0.5:0.95增加了28%,检测性能提升明显. 由于Faster R-CNN本身缺乏轻量化设计,在引入DMT后的参数量增加了24%,处理速度下降了23%,与精度的提升相比,这一代价仍处于可接受范围. 综上,在不同架构模型中,DMT能够在保持轻量化的前提下取得稳定且显著的性能提升,证明所提框架具有良好的通用性与实用价值.

表 4   动态多尺度与纹理感知框架在不同基准算法上的性能对比

Tab.4  Performance comparison of dynamic multi-scale and texture-aware framework on different baseline algorithms

模型P/%R/%F1/%mAP@0.5/%mAP@0.5:0.95/%np/106rf/(帧·s−1)
YOLOv885.672.378.485.146.13.1677.50
DMT-YOLOv891.181.285.989.150.23.1976.20
Faster R-CNN (ResNet50+FPN)51.962.756.861.028.841.3422.30
DMT-Faster R-CNN (ResNet50+FPN)64.875.669.871.136.551.3717.10
YOLOv5n82.171.681.381.845.81.7759.72
DMT-YOLOv5n87.081.284.085.948.51.8155.26

新窗口打开| 下载CSV


为了进一步验证所提模块的有效性,选取4种性能优于YOLOv8的代表性检测算法进行对比,分别为YOLOv9、YOLOv10、YOLOv11和RT-DETR[28]. 其中YOLOv9–11代表YOLO系列在不同阶段的最优成果;RT-DETR是近年来提出的端到端检测架构,具有较强的全局建模能力. 为了保证实验的公平性,所有算法均基于同一YOLO官方框架实现,均采用早停策略:当连续30个迭代轮次未取得性能提升时自动终止训练. 设置批次大小为2,学习率及其他超参数均保持默认配置,使用YOLO系列默认训练策略. 如表5所示,原始YOLOv8在多项指标上均明显弱于新算法,这表明YOLOv8在复杂缺陷检测任务中存在性能瓶颈. 在引入DMT后,DMT-YOLOv8在精确率、召回率和F1上均实现显著提升,不仅全面超越了基线YOLOv8,也整体反超其他对比算法,取得最优结果.

表 5   不同算法在轨道缺陷检测任务中的性能对比

Tab.5  Comparison of detection performance among different algorithms on railway defect detection %

算法PRF1mAP@0.5mAP@0.5:0.95
YOLOv885.672.378.485.146.1
YOLOv986.282.684.487.649.7
YOLOv1084.481.283.185.147.7
YOLOv1188.174.480.683.544.7
RT-DETR90.179.984.786.147.5
DMT-YOLOv891.181.985.989.150.2

新窗口打开| 下载CSV


为了进一步展示各算法的检测性能,选取YOLOv8、DMT-YOLOv8以及表现较优的YOLOv9和RT-DETR,绘制PR曲线,结果如图6所示. 可以看出,4种算法的PR曲线均呈现出随召回率提升而精度逐渐下降的趋势. 其中YOLOv8的曲线在下降过程中波动较大,且整体精度低于其他方法,可见该算法识别小目标与模糊目标的能力较弱;相比之下,YOLOv9与RT-DETR在中高召回率区间的曲线更为平稳,能够在保持较高召回率的同时维持较优的检测精度,整体表现优于YOLOv8;DMT-YOLOv8的PR曲线整体优于对比算法,当召回率约为80%时,DMT-YOLOv8的精确率达到82.3%,明显优于YOLOv9(P=81.9%)、RT-DETR(P=72.9%)和YOLOv8(P=70.9%). 说明在中高召回区域,DMT-YOLOv8能够在保持召回率稳定的同时获得更高的检测精度. 该实验结果表明,所提架能够有效提升模型对复杂缺陷的识别能力,使模型在保持较高召回率的同时维持较高精度.

图 6

图 6   不同检测算法PR曲线对比

Fig.6   Comparison of PR curves of different detection algorithms


综合以上2个实验结果能够看出,DMT能够有效弥补YOLOv8在小目标与模糊目标检测方面的不足,提升算法整体检测性能.

2.5. 可视化与泛化实验

选取YOLOv8、YOLOv9和RT-DETR为对比算法,在测试数据集上对所有检测结果进行可视化,为了便于展示与对比,将图像进行适当的剪裁处理. 考虑篇幅限制,仅选取部分具有代表性的样例进行展示,如图7所示. 从可视化结果可以看出,YOLOv8在处理小目标和模糊目标时存在一定局限性:在第一第二张图中,出现小目标和模糊目标漏检的情况;在第三张和第四张图中,出现模糊目标误检的情况. 相比之下,YOLOv9和RT-DETR的特征提取能力有所提升,但算法缺乏对于模糊缺陷边缘信息的有效处理,在低对比度场景中仍容易出现误检,在4个示例中均可观察到类似情况. DMT-YOLOv8模型能够在复杂条件下准确识别小目标和模糊目标,整体检测质量显著优于其他对比模型. 该结果进一步验证了DMT在增强模型小目标感知能力和模糊目标识别能力方面的有效性与实用价值.

图 7

图 7   不同算法在轨道缺陷数据集上的可视化检测结果

Fig.7   Visual detection results of different algorithms on railway defect dataset


选取真实轨道场景缺陷数据集Rail-1118[29]进行实验,验证DMT在真实场景中的适用性与鲁棒性. 如表6所示,DMT-YOLOv8相较于基准YOLOv8在各项评估指标上均取得稳定提升. 其中召回率提升6.7个百分点,表明所设计模块能够有效增强对小目标与模糊缺陷的检测能力. 如图8所示,在真实轨道场景中,DMT同样能够显著提升模型对小目标和模糊缺陷的识别能力,减少错检与漏检. 综上所述,本研究提出的框架在真实复杂场景下仍表现出较强的检测性能与良好的泛化能力.

表 6   Rail-1118数据集上2种检测算法的性能对比结果

Tab.6  Performance comparison results of two detection algorithms on Rail-1118 dataset %

算法PRF1mAP@0.5mAP@0.5:0.95
YOLOv858.168.763.060.734.6
DMT-YOLOv861.475.467.762.636.5

新窗口打开| 下载CSV


图 8

图 8   Rail-1118数据集上2种算法的可视化检测结果

Fig.8   Visual detection results of two algorithms on Rail-1118 dataset


2.6. 错误样例分析

为了全面地评估所提模型的性能,对检测失败样例进行分析,将主要错误类型分为4个类别,典型结果如图9所示. 第一列展示的错误样例主要源于数据集标注精度不足,模型实际上正确识别出缺陷,但与标注结果不一致;第二列样例对应连续性缺陷,模型因难以准确划定分界线而出现多检或误检,这是数据集标注准则不一致导致的错误;第三和第四列反映了模型在小目标和模糊目标检测方面仍存在不足,对于部分极小目标和高度模糊缺陷,仍存在漏检现象.

图 9

图 9   所提模型检测错误样例

Fig.9   Examples of detection failures of proposed model


对错误样例进行分类统计,结果如表7所示. 由原标注漏标和连续型缺陷误检导致的错误占57.7%,这说明数据集在质量和标注一致性方面存在不足;小目标漏检导致的错误占11.5%,说明DMT的小目标建模效果较好;模糊目标的漏检、误检占30.8%,仍属于较为突出的问题,在未来可进一步强化针对模糊目标的建模能力.

表 7   所提模型检测错误类型统计

Tab.7  Statistics of detection error types for proposed model

错误描述数量占比/%
原标注漏标830.8
连续型缺陷误检726.9
小目标漏检311.5
模糊目标漏检、误检830.8

新窗口打开| 下载CSV


综上所述,轨道缺陷检测领域现有数据集在标注精度和标注一致性方面仍有提升空间,DMT及其设计模块亦存在进一步改进空间.

3. 结 语

针对轨道缺陷检测中小目标易漏检与模糊缺陷易误检的问题,本研究提出融合多尺度与纹理边缘感知的检测框架,设计3个特征增强模块,构造新的轨道缺陷检测算法. 在真实轨道缺陷数据集上的实验结果表明,所设计模块能够有效提升模型在小目标与模糊缺陷场景下的检测能力,并在多种基准算法中展现出良好的通用性. 泛化实验进一步验证了所提框架在领域中的适应性与鲁棒性. 未来的工作计划从构建更高质量的数据集,提升标注一致性,引入跨模态信息以及设计更高效的特征建模机制等方向展开.

参考文献

董超群, 汪战, 廖平, 等

轻量化YOLOv5s-OCG的轨枕裂纹检测算法

[J]. 浙江大学学报: 工学版, 2025, 59 (9): 1838- 1845

[本文引用: 1]

DONG Chaoqun, WANG Zhan, LIAO Ping, et al

Lightweight YOLOv5s-OCG rail sleeper crack detection algorithm

[J]. Journal of Zhejiang University: Engineering Science, 2025, 59 (9): 1838- 1845

[本文引用: 1]

吴永军, 崔灿, 何永福

基于语义增广与YOLOv8的钢轨表面缺陷检测方法

[J]. 铁道科学与工程学报, 2024, 21 (9): 3864- 3875

DOI:10.19713/j.cnki.43-1423/u.T20231930      [本文引用: 1]

WU Yongjun, CUI Can, HE Yongfu

Rail surface defect detection based on semantic augmentation and YOLOv8

[J]. Journal of Railway Science and Engineering, 2024, 21 (9): 3864- 3875

DOI:10.19713/j.cnki.43-1423/u.T20231930      [本文引用: 1]

GIBERT X, PATEL V M, CHELLAPPA R

Deep multitask learning for railway track inspection

[J]. IEEE Transactions on Intelligent Transportation Systems, 2017, 18 (1): 153- 164

DOI:10.1109/TITS.2016.2568758      [本文引用: 1]

金学松, 李霞, 李伟, 等

铁路钢轨波浪形磨损研究进展

[J]. 西南交通大学学报, 2016, 51 (2): 264- 273

DOI:10.3969/j.issn.0258-2724.2016.02.006      [本文引用: 1]

JIN Xuesong, LI Xia, LI Wei, et al

Review of rail corrugation progress

[J]. Journal of Southwest Jiaotong University, 2016, 51 (2): 264- 273

DOI:10.3969/j.issn.0258-2724.2016.02.006      [本文引用: 1]

高天赐, 史一帆, 江乐鹏, 等

基于瓦片编码网络的钢轨焊缝几何不平顺识别

[J]. 铁道科学与工程学报, 2025, 22 (5): 2346- 2354

DOI:10.19713/j.cnki.43-1423/u.T20241206      [本文引用: 1]

GAO Tianci, SHI Yifan, JIANG Lepeng, et al

A tile-coding-based neural network for rail weld irregularity identification

[J]. Journal of Railway Science and Engineering, 2025, 22 (5): 2346- 2354

DOI:10.19713/j.cnki.43-1423/u.T20241206      [本文引用: 1]

PHUSAKULKAJORN W, ZENG Y, LI Z, et al

Unsupervised representation learning for monitoring rail infrastructures with high-frequency moving vibration sensors

[J]. IEEE Transactions on Intelligent Transportation Systems, 2025, 26 (8): 12746- 12760

DOI:10.1109/TITS.2025.3557712      [本文引用: 1]

CHEN Z, WANG Q, HE Q, et al

CUFuse: camera and ultrasound data fusion for rail defect detection

[J]. IEEE Transactions on Intelligent Transportation Systems, 2022, 23 (11): 21971- 21983

DOI:10.1109/TITS.2022.3189677      [本文引用: 1]

NI X, MA Z, LIU J, et al

Attention network for rail surface defect detection via consistency of intersection-over-union(IoU)-guided center-point estimation

[J]. IEEE Transactions on Industrial Informatics, 2022, 18 (3): 1694- 1705

DOI:10.1109/TII.2021.3085848      [本文引用: 2]

姚宗伟, 杨宏飞, 胡际勇, 等

基于机器视觉和卷积神经网络的轨道表面缺陷检测方法

[J]. 铁道学报, 2021, 43 (4): 101- 107

DOI:10.3969/j.issn.1001-8360.2021.04.013      [本文引用: 1]

YAO Zongwei, YANG Hongfei, HU Jiyong, et al

Track surface defect detection method based on machine vision and convolutional neural network

[J]. Journal of the China Railway Society, 2021, 43 (4): 101- 107

DOI:10.3969/j.issn.1001-8360.2021.04.013      [本文引用: 1]

徐心慈, 师秀霞, 耿晨歌, 等

基于深度学习的铁轨图像识别技术

[J]. 铁道科学与工程学报, 2024, 21 (12): 5232- 5241

DOI:10.19713/j.cnki.43-1423/u.T20240411     

XU Xinci, SHI Xiuxia, GENG Chenge, et al

Rail image recognition technology based on deep learning

[J]. Journal of Railway Science and Engineering, 2024, 21 (12): 5232- 5241

DOI:10.19713/j.cnki.43-1423/u.T20240411     

YE J, STEWART E, CHEN Q, et al

Deep learning and laser-based 3-D pixel-level rail surface defect detection method

[J]. IEEE Transactions on Instrumentation and Measurement, 2023, 72: 2513612

DOI:10.1109/tim.2023.3272033      [本文引用: 1]

刘子豪, 张佳欣, 薛峰, 等

基于改进YOLO-v8的精密管件表面缺陷检测方法

[J]. 浙江大学学报: 工学版, 2025, 59 (7): 1514- 1522

[本文引用: 1]

LIU Zihao, ZHANG Jiaxin, XUE Feng, et al

Surface defect detection method of precision pipe fittings based on improved YOLO-v8

[J]. Journal of Zhejiang University: Engineering Science, 2025, 59 (7): 1514- 1522

[本文引用: 1]

ZHAO Q, FANG S, LI Y, et al

Track defect detection based on improved YOLOv5s

[J]. IEEE Transactions on Industrial Informatics, 2025, 21 (4): 3346- 3355

DOI:10.1109/TII.2024.3523593     

WANG B, ZHOU W, YAN W, et al

PENet-KD: progressive enhancement network via knowledge distillation for rail surface defect detection

[J]. IEEE Transactions on Instrumentation and Measurement, 2023, 72: 5032811

DOI:10.1109/tim.2023.3330222     

ZHANG Y, LIU C

Real-time pavement damage detection with damage shape adaptation

[J]. IEEE Transactions on Intelligent Transportation Systems, 2024, 25 (11): 18954- 18963

DOI:10.1109/TITS.2024.3416508      [本文引用: 1]

KUMAR A, HARSHA S P

A systematic literature review of defect detection in railways using machine vision-based inspection methods

[J]. International Journal of Transportation Science and Technology, 2025, 18: 207- 226

DOI:10.1016/j.ijtst.2024.06.006      [本文引用: 1]

QIU Y, LIU H, LIU J, et al

Region and edge-aware network for rail surface defect segmentation

[J]. IEEE Transactions on Instrumentation and Measurement, 2024, 73: 3522213

DOI:10.1109/tim.2024.3406815      [本文引用: 1]

HUSSAIN M

YOLO-v1 to YOLO-v8, the rise of YOLO and its complementary nature toward digital manufacturing and industrial defect detection

[J]. Machines, 2023, 11 (7): 677

DOI:10.3390/machines11070677      [本文引用: 1]

WANG Q, WU B, ZHU P, et al. ECA-net: efficient channel attention for deep convolutional neural networks [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2020: 11531–11539.

WU Y, QIANG F, ZHOU W, et al

PFCNet: enhancing rail surface defect detection with pixel-aware frequency conversion networks

[J]. IEEE Signal Processing Letters, 2025, 32: 606- 610

DOI:10.1109/LSP.2025.3525855      [本文引用: 1]

LIU Y, XIAO H, XU J, et al

A rail surface defect detection method based on pyramid feature and lightweight convolutional neural network

[J]. IEEE Transactions on Instrumentation and Measurement, 2022, 71: 5009510

DOI:10.1109/tim.2022.3165287      [本文引用: 1]

LIN T Y, DOLLÁR P, GIRSHICK R, et al. Feature pyramid networks for object detection [C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Honolulu: IEEE, 2017: 936–944.

LI X, XU X, YANG H

A road crack detection model integrating GLMANet and EFPN

[J]. IEEE Transactions on Intelligent Transportation Systems, 2024, 25 (11): 18211- 18223

DOI:10.1109/TITS.2024.3432995     

YU F, KOLTUN V. Multi-scale context aggregation by dilated convolutions [EB/OL]. (2016−04−30)[2025−06−02]. https://arxiv.org/pdf/1511.07122.

[本文引用: 1]

张振利, 胡新凯, 李凡, 等

基于CNN和Efficient Transformer的多尺度遥感图像语义分割算法

[J]. 浙江大学学报: 工学版, 2025, 59 (4): 778- 786

[本文引用: 1]

ZHANG Zhenli, HU Xinkai, LI Fan, et al

Semantic segmentation algorithm for multiscale remote sensing images based on CNN and Efficient Transformer

[J]. Journal of Zhejiang University: Engineering Science, 2025, 59 (4): 778- 786

[本文引用: 1]

WANG T, ZHANG Z, TSUI K L

A hierarchical transfer-generative framework for automating multianalytical tasks in rail surface defect inspection

[J]. IEEE Internet of Things Journal, 2024, 11 (12): 21513- 21526

DOI:10.1109/JIOT.2024.3374751      [本文引用: 1]

BONO F M. Railway computer vision model [DB/OL]. [2025−06−02]. https://universe.roboflow.com/francesco-morgan-bono-cntpn/railway-yyihi.

[本文引用: 1]

ZHAO Y, LV W, XU S, et al. DETRs beat YOLOs on real-time object detection [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2024: 16965–16974.

[本文引用: 1]

Roboflow Universe. Rail surface defects computer vision dataset [DB/OL]. [2025−11−03]. https://universe.roboflow.com/dataset-aq7x0/rail-surface-defects-flrty.

[本文引用: 1]

/

〈 〉