Please wait a minute...
浙江大学学报(工学版)  2026, Vol. 60 Issue (9): 1901-1911    DOI: 10.3785/j.issn.1008-973X.2026.09.007
计算机技术、自动控制技术     
基于YOLOv11n的改进铁路工人安全穿戴检测模型
武晓春(),李梓宁
兰州交通大学 自动化与电气工程学院,甘肃 兰州 730070
Improved railway worker safety wear detection model based on YOLOv11n
Xiaochun WU(),Zining LI
School of Automation and Electrical Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China
 全文: PDF(5035 KB)   HTML
摘要:

针对铁路作业场景较复杂,安全穿戴远景目标较小及边缘计算设备资源有限等问题,提出VMS-YOLOv11n的铁路安全穿戴检测模型. 设计VABlock特征提取模块,替换主干网络及颈部网络中的C3k2模块,在保留关键特征的同时减少计算冗余. 设计MSF多尺度特征融合结构,通过融合多级特征加权策略与双向特征金字塔网络,提高铁路复杂场景下的小目标检测精度. 构建空间金字塔特征融合结构(SE2SPPF),在增强小目标特征学习能力的同时提高模型检测能力. 实验结果表明,相较于基线YOLOv11n,改进模型的精确率为98%,召回率为92.5%,平均精度均值为97.5%,分别提高了2.9%、2.5%和2.7%. 参数量为2.45×106,与基线模型相比降低了5%. 改进后的模型在铁路安全穿戴检测中的效果良好,有效平衡了检测精度和速度.

关键词: 铁路安全穿戴检测YOLO复杂场景小目标特征提取    
Abstract:

A VMS-YOLOv11n model for railroad safety wear detection was proposed aiming at the problems of more complex railroad operation scenario, smaller safety wear vision targets and limited resource of edge computing devices. The VABlock feature extraction module was designed to replace the C3k2 module in the backbone network as well as the neck network to reduce computational redundancy while retaining key features. Then the MSF multi-scale feature fusion structure was designed to improve the small target detection accuracy in complex railroad scene by fusing the multilevel feature weighting strategy with the bidirectional feature pyramid network. The spatial pyramid feature fusion structure (SE2SPPF) was constructed to enhance the small target feature learning capability while improving the model detection capability. The experimental results showed that the improved model had precision rate of 98%, recall rate of 92.5%, and average precision mean of 97.5%, which were 2.9%, 2.5% and 2.7% higher, respectively compared with the baseline YOLOv11n. The number of parameters is 2.45×106, which is 5% lower than the baseline. The improved model works well in railroad safety wear detection, effectively balancing detection accuracy and speed.

Key words: wear detection for railroad safety    YOLO    complex scenario    small target    feature extraction
收稿日期: 2025-09-26 出版日期: 2026-07-20
CLC:  TP 391  
基金资助: 国家自然科学基金资助项目(61661027);中央引导地方科技发展资金资助项目(24ZYQA044);甘肃省重点研发计划资助项目(22YF7GA141).
作者简介: 武晓春(1973—),女,教授,硕士,从事交通信息工程及控制研究. orcid.org/0009-0005-9874-452X. E-mail:369038806@qq.com
服务  
把本文推荐给朋友
加入引用管理器
E-mail Alert
作者相关文章  
武晓春
李梓宁

引用本文:

武晓春,李梓宁. 基于YOLOv11n的改进铁路工人安全穿戴检测模型[J]. 浙江大学学报(工学版), 2026, 60(9): 1901-1911.

Xiaochun WU,Zining LI. Improved railway worker safety wear detection model based on YOLOv11n. Journal of ZheJiang University (Engineering Science), 2026, 60(9): 1901-1911.

链接本文:

https://www.zjujournals.com/eng/CN/10.3785/j.issn.1008-973X.2026.09.007        https://www.zjujournals.com/eng/CN/Y2026/V60/I9/1901

图 1  VMS-YOLO模型的结构
图 2  VoV-GSCSP的结构
图 3  AKConv的结构
图 4  VABlock的结构
图 5  GSConv的结构
图 6  MSF模块的结构
图 7  SE2SPPF模块的结构
图 8  铁路工人安全穿戴检测数据集的示例
参数说明
操作系统Ubuntu 22.04
运行内存120 GB
显卡GeForce RTX 4090
显存24 GB
CPU16 vCPU Intel(R) Xeon(R) Gold 6430
深度学习框架Pytorch 2.1.0
CUDA版本12.1
Python版本3.10
表 1  实验环境的参数设置
图 9  训练损失函数曲线
图 10  PConv所引入SSFF结构中的不同位置
模块卷积
位置
P/%R/%mAP50/
%
mAP50:95/
%
FLOPs/109Np/106
基线模型94.490.394.865.46.32.58
MSF1A95.590.195.566.76.32.95
MSF2B95.889.995.466.46.32.95
MSFA+B96.490.896.266.96.32.96
表 2  不同改进方式的MSF模块性能结果对比
模型P/%R/%mAP50/
%
mAP50:95/
%
FLOPs/109Np/106
基线模型94.490.394.865.46.32.58
BiFPN95.190.195.366.96.31.92
AFPN95.691.295.766.88.82.65
GFPN95.290.895.667.18.23.66
MSF96.490.896.266.96.32.96
表 3  不同颈部网络结构的性能对比
方法P/%R/%mAP50/
%
mAP50:95/
%
FLOPs/109Np/106
基线模型94.490.394.865.46.32.58
DSConv94.489.394.865.34.31.99
DWConv94.089.994.965.54.11.92
GhostConv95.690.495.166.34.82.25
ODConv95.889.695.366.49.54.60
GSConv95.989.995.366.54.82.25
表 4  不同类型卷积的性能对比
图 11  AKConv所引入VoVGSCSP结构中的不同位置
模块AKConv位置P/%R/%mAP50/%mAP50:95/%FLOPs/109Np/106
基线模型94.490.394.865.46.32.58
VABlock1A95.186.094.862.85.42.49
VABlock2B94.789.694.066.15.42.49
VABlock3C93.686.295.660.36.02.93
VABlock4D95.589.192.760.95.32.59
VABlockE96.289.495.666.74.82.19
表 5  不同结构的VABlock模块性能对比
图 12  引入VABlock前、后对遮挡图像的特征提取热力图对比
模型VABlockMSFSE2SPPFP/%R/%mAP50/%mAP50:95/%FLOPs/109Np/106
YOLOv11n94.490.394.865.46.32.58
V-YOLO96.289.495.666.76.02.19
M-YOLO96.490.896.266.97.72.96
S-YOLO96.389.395.966.76.32.62
VM-YOLO96.689.596.367.17.12.41
VS-YOLO96.491.696.567.36.02.22
MS-YOLO96.891.496.767.57.72.99
VMS-YOLO98.092.597.567.87.12.45
表 6  消融实验结果
模型P/%R/%mAP50/%mAP50:95/%FLOPs/109Np/106v/(帧·s?1)
Faster R-CNN69.754.462.545.0948.128.347
SSD85.584.390.859.86.23.8131
Mamba-YOLO95.590.895.766.913.65.9862
GhostNetP293.484.991.856.97.31.45167
YOLO-Worldv295.391.795.666.99.63.53152
RT-DETR95.792.696.570.596.625.66125
YOLOv5n95.890.295.466.87.82.18230
YOLOv8n94.685.591.257.98.13.00278
YOLO11n94.490.394.865.45.52.58197
VMS-YOLO98.092.597.567.85.42.45238
表 7  不同模型的检测性能对比结果
图 13  模型改进前、后的检测效果对比
模型vesthelmetworkerFLOPs/109
Np/106
P//%R//%mAP50/%P/%R/%mAP50/%P/%R/%mAP50/%
YOLOv11n93.997.098.495.976.387.693.497.798.56.32.58
VMS-YOLO96.899.599.898.578.292.898.599.899.85.42.45
表 8  改进前、后的各目标检测结果对比
模型P/%R/%mAP50/%mAP50:95/%FLOPs/109Np/106v/(帧·s?1)
YOLOv11n90.183.387.158.46.32.58198
VMS-YOLO93.390.392.063.65.42.45236
表 9  泛化实验结果
图 14  泛化实验的可视化结果对比
1 方文珊 《铁路轨道工程施工质量检测技术规程》主要技术标准研究[J]. 铁路工程技术与经济, 2025, 40 (1): 11- 14
FANG Wenshan Research on the major technical regulations in technical code for constructional quality detection of railway track[J]. Railway Engineering Technology and Economy, 2025, 40 (1): 11- 14
doi: 10.20262/j.cnki.issn.2097-6186.2025.01.03
2 周瑶, 周石 基于YOLOv5改进的铁路工人安全帽检测算法研究[J]. 计算机测量与控制, 2024, 32 (3): 71- 78
ZHOU Yao, ZHOU Shi Research on the detection algorithm of railway worker’s hard hat based on YOLOv5 improvement[J]. Computer Measurement and Control, 2024, 32 (3): 71- 78
doi: 10.16526/j.cnki.11-4762/tp.2024.03.011
3 REN S, HE K, GIRSHICK R, et al Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39 (6): 1137- 1149
doi: 10.1109/TPAMI.2016.2577031
4 HE K, GKIOXARI G, DOLLÁR P, et al. Mask R-CNN [C]//Proceedings of the IEEE International Conference on Computer Vision. Venice: IEEE, 2017: 2980–2988.
5 REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection [C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2016: 779-788.
6 LIU W, ANGUELOV D, ERHAN D, et al. SSD: single hot multiBox detector [C]//European Conference on Computer Vision. Cham: Springer, 2016: 21-37.
7 LING L, XU S, WEI L, et al Fsd-detr: casting surface defect detection based on improved RT-DETR[J]. Journal of Real-Time Image Processing, 2025, 22 (4): 135
doi: 10.1007/s11554-025-01714-x
8 冯勇, 杨思卓, 徐红艳 基于YOLO v8的轻量化安全帽佩戴检测算法[J]. 计算机应用, 2024, 44 (Suppl.2): 251- 256
FENG Yong, YANG Sizhuo, XU Hongyan Lightweight safety helmet wearing detection algorithm based on YOLO v8[J]. Journal of Computer Applications, 2024, 44 (Suppl.2): 251- 256
9 SHAN C, LIU H, YU Y Research on improved algorithm for helmet detection based on YOLOv5[J]. Scientific Reports, 2023, 13: 18056
doi: 10.1038/s41598-023-45383-x
10 徐壮, 钱育蓉, 颜丰 GCW-YOLOv8n: 轻量级安全帽佩戴检测算法[J]. 计算机工程与应用, 2025, 61 (3): 144- 154
XU Zhuang, QIAN Yurong, YAN Feng GCW-YOLOv8n: lightweight safety helmet wearing detection algorithm[J]. Computer Engineering and Applications, 2025, 61 (3): 144- 154
11 冯爽, 王万齐, 杨文, 等 基于改进RT-DETR的铁路施工场景下人员安全穿戴检测[J]. 铁道学报, 2025, 47 (2): 92- 101
FENG Shuang, WANG Wanqi, YANG Wen, et al Safety wear detection for personnel in railway construction scenarios based on improved RT-DETR[J]. Journal of the China Railway Society, 2025, 47 (2): 92- 101
doi: 10.3969/j.issn.1001-8360.2025.02.010
12 KHANAM R, HUSSAIN M. YOLOv11: an overview of the key architectural enhancements [EB/OL]. [2025-09-10]. https://arxiv.org/abs/2410.17725.
13 LI H, LI J, WEI H, et al. Slim-neck by GSConv: a lightweight-design for real-time detector architectures [EB/OL]. [2025-09-10]. https://arxiv.org/abs/2206.02424.
14 ZHANG X, SONG Y, SONG T, et al. AKConv: convolutional kernel with arbitrary sampled shapes and arbitrary number of parameters [EB/OL]. [2025-09-10]. https://arxiv.org/abs/2311.11587v1.
15 KANG M, TING C M, TING F F, et al ASF-YOLO: a novel YOLO model with attentional scale sequence fusion for cell instance segmentation[J]. Image and Vision Computing, 2024, 147: 105057
doi: 10.1016/j.imavis.2024.105057
16 CHEN J, KAO S H, HE H, et al. Run, don’t walk: chasing higher FLOPS for faster neural networks [C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Vancouver: IEEE, 2023: 12021–12031.
17 程德强, 姬广凯, 张皓翔, 等 基于多粒度融合和跨尺度感知的跨模态行人重识别[J]. 通信学报, 2025, 46 (1): 108- 123
CHENG Deqiang, JI Guangkai, ZHANG Haoxiang, et al Cross-modality person re-identification based on multi-granularity fusion and cross-scale perception[J]. Journal on Communications, 2025, 46 (1): 108- 123
18 NARAYANAN M. SENetV2: aggregated dense layer for channelwise and global representations [EB/OL]. [2025-09-10]. https://arxiv.org/abs/2311.10807.
19 MAZUROV M. Railroad worker detection dataset [DB/OL]. [2025-05-15]. https://www.kaggle.com/datasets/johnsmith/shop-ping-trends.
20 TAN M, PANG R, LE Q V. EfficientDet: scalable and efficient object detection [C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2020: 10778–10787.
21 YANG G, LEI J, ZHU Z, et al. AFPN: asymptotic feature pyramid network for object detection [C]//Proceedings of the IEEE International Conference on Systems, Man, and Cybernetics. Honolulu: IEEE, 2024: 2184–2189.
22 JIANG Y, TAN Z, WANG J, et al. GiraffeDet: a heavy-neck paradigm for object detection [EB/OL]. [2025-09-10]. https://arxiv.org/abs/2202.04256.
23 DO NASCIMENTO M G, PRISACARIU V, FAWCETT R. DSConv: efficient convolution operator [C]//Proceedings of the IEEE/CVF International Conference on Computer Vision. Seoul: IEEE, 2020: 5147–5156.
24 HAN Q, FAN Z, DAI Q, et al. On the connection between local attention and dynamic depth-wise convolution [EB/OL]. [2025-09-10]. https://arxiv.org/abs/2106.04263.
25 CAO J, BAO W, SHANG H, et al GCL-YOLO: a GhostConv-based lightweight YOLO network for UAV small object detection[J]. Remote Sensing, 2023, 15 (20): 4932
doi: 10.3390/rs15204932
26 LI C, ZHOU A, YAO A. Omni-dimensional dynamic convolution [EB/OL]. [2025-09-10]. https://arxiv.org/abs/2209.07947.
27 WANG Z, LI C, XU H, et al. Mamba YOLO: a simple baseline for object detection with state space model [EB/OL]. [2025-09-10]. https://arxiv.org/abs/2406.05835.
28 TANG Y, HAN K, GUO J, et al GhostNetv2: enhance cheap operation with long-range attention[J]. Advances in Neural Information Processing Systems, 2022, 35: 9969- 9982
doi: 10.52202/068431-0724
29 CHENG T, SONG L, GE Y, et al. YOLO-world: real-time open-vocabulary object detection [C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2024: 16901–16911.
[1] 连远锋,范树玉,张本哲,王森. 融合几何约束特征提取与多奖励协同优化的仿生步态控制[J]. 浙江大学学报(工学版), 2026, 60(9): 1862-1871.
[2] 火久元,姜灏,常琛. 基于改进EAST的高速公路交通标识文本检测[J]. 浙江大学学报(工学版), 2026, 60(9): 1881-1889.
[3] 李珺,丁彬彬,史维娟,杨琳. 基于多尺度特征聚合的航拍图像检测算法[J]. 浙江大学学报(工学版), 2026, 60(9): 1890-1900.
[4] 柳长源,赵海健,吴海滨,刘佳伟. 面向自动驾驶的轻量化交警手势识别方法[J]. 浙江大学学报(工学版), 2026, 60(8): 1678-1685.
[5] 肖剑,杨小苑,何昕泽,陈林,胡欣. 基于全局信息感知的轻量级螺纹钢表面缺陷检测算法[J]. 浙江大学学报(工学版), 2026, 60(7): 1438-1451.
[6] 吴佰靖,闫光辉,马龙,程文鑫,黄亚宁. 边缘感知和跨尺度特征增强的小目标水漂垃圾检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1452-1463.
[7] 于鑫淼,夏楠,江佳鸿,郝子莹,把云胜. 基于多尺度特征相似性匹配的低照度目标检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1464-1474.
[8] 孟昱煜,马银宝,火久元. 基于无人机航拍图像的实时车辆检测算法[J]. 浙江大学学报(工学版), 2026, 60(7): 1599-1610.
[9] 张乃洲,赵云超,曹薇,张啸剑. 基于多视图跨模态特征融合的图像描述生成[J]. 浙江大学学报(工学版), 2026, 60(6): 1205-1212.
[10] 于天河,王文龙,刘镛,杨壮壮,侯善冲. 改进的有雾图像中被遮挡车辆及行人识别算法[J]. 浙江大学学报(工学版), 2026, 60(4): 738-750.
[11] 宋耀莲,彭驰,唐菁敏,赵宣植,虞贵财. 基于融合注意力机制的光学遥感图像小目标检测算法[J]. 浙江大学学报(工学版), 2026, 60(4): 763-771.
[12] 田卫,周菻鈜,李欣阳,王建明,黄余康. 基于改进YOLOv8的3D打印混凝土表观缺陷检测方法[J]. 浙江大学学报(工学版), 2026, 60(4): 833-843.
[13] 李彬彬,张超,覃涛,陈昌盛,刘兴艳,杨靖. 面向光伏电站建设的移动端人体跌倒检测方法[J]. 浙江大学学报(工学版), 2026, 60(3): 546-555.
[14] 雒伟群,陆敬蔚,吴佳缔,梁钰迎,申传鹏,朱睿. 藏区高原典型环境地形目标的轻量化检测模型[J]. 浙江大学学报(工学版), 2026, 60(3): 594-603.
[15] 孟昱煜,孔垂乐,火久元,武泽宇. 重构YOLOv11的无人机小目标检测算法[J]. 浙江大学学报(工学版), 2026, 60(2): 303-312.