基于轻量化重参数化YOLOv11的铁轨障碍物检测方法
DOI:
作者:
作者单位:

华东交通大学

作者简介:

通讯作者:

中图分类号:

基金项目:

基于深度学习的高铁轨道状态监测与病害预测研究及应用


A Lightweight Re-parameterized YOLOv11-Based Method for Railway Obstacle Detection
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对铁轨障碍物检测模型在边缘设备部署时面临的模型体积大、计算复杂度高、精度损失显著等问题,本文提出一种轻量化重参数化网络YOLOv11s-Slim-Rep。该网络通过两项核心改进实现精度与效率的协同优化:将网络宽度缩放因子由0.50调整为0.35,使模型大小压缩44.8%,参数量减少45.4%;采用RepConvCustom模块替换全部stride=2下采样卷积层,通过结构重参数化技术融合为单个卷积,实现推理阶段零额外开销。在NewData和Railway两个铁路轨道障碍物数据集上,与YOLOv11s基线模型相比,mAP50仅下降约1.1%,模型压缩近45%;与YOLOv11n相比,在复杂场景下mAP50提升4.7%,跨数据集mAP50波动仅0.6%。所提方法在显著压缩模型的同时保持了较高的检测精度,展现出良好的边缘设备部署潜力。

    Abstract:

    To address the challenges of large model size, high computational complexity, and notable accuracy degradation when deploying railway track obstacle detection models on edge devices, this paper proposes a lightweight re-parameterized network, YOLOv11s-Slim-Rep. The network achieves synergistic optimization of accuracy and efficiency through two core improvements: adjusting the network width scaling factor from 0.50 to 0.35, resulting in a 44.8% reduction in model size and a 45.4% decrease in parameter count; and replacing all stride=2 downsampling convolutional layers with a RepConvCustom module, which is fused into a single convolution via structural re-parameterization, incurring zero additional inference overhead. On two railway track obstacle datasets (NewData and Railway), compared to the YOLOv11s baseline, the mAP50 decreases by only approximately 1.1% while achieving nearly 45% model compression. Compared to YOLOv11n, it improves mAP50 by 4.7% in complex scenarios, with a cross-dataset mAP50 fluctuation of only 0.6%. The proposed method maintains high detection accuracy while significantly compressing the model, demonstrating strong potential for edge deployment.

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2026-03-30
  • 最后修改日期:2026-05-08
  • 录用日期:2026-05-12
  • 在线发布日期: 2026-06-24
  • 出版日期:
关闭