改进YOLO11检测铁路输电线异物,提升精度与速度。
MRS-YOLO Railroad Transmission Line Foreign Object Detection Based on Improved YOLO11 and Channel Pruning
- 引入多尺度特征融合模块增强小目标检测能力
- mAP50达94.8%,mAP50:95达86.4%,精度显著提升
- 通过通道剪枝降低44.2%参数量,适合部署于边缘设备
针对铁路环境下输电线异物检测存在的漏检、误检及效率低问题,提出基于YOLO11的改进算法MRS-YOLO。首先设计多尺度自适应核深度特征融合(MAKDF)模块,与C3k2模块结合形成C3k2_MAKDF,提升对不同尺寸和形状异物的特征提取能力;其次构建新型重校准特征融合金字塔网络(RCFPN)作为颈部结构,强化多层级特征融合与利用;再设计基于空间与通道重构的检测头(SC_Detect),进一步提升检测性能;最后采用通道剪枝技术减少模型冗余,使参数量和GFLOPs分别降低44.2%和17.5%。实验表明,MRS-YOLO在铁路输电线异物检测任务中mAP50达到94.8%,mAP50:95为86.4%,较基线分别提升0.7和2.3个百分点,同时显著提升检测效率,具备实际部署价值。
原文摘要 · Abstract (English)
Aiming at the problems of missed detection, false detection and low detection efficiency in transmission line foreign object detection under railway environment, we proposed an improved algorithm MRS-YOLO based on YOLO11. Firstly, a multi-scale Adaptive Kernel Depth Feature Fusion (MAKDF) module is proposed and fused with the C3k2 module to form C3k2_MAKDF, which enhances the model's feature extraction capability for foreign objects of different sizes and shapes. Secondly, a novel Re-calibration Feature Fusion Pyramid Network (RCFPN) is designed as a neck structure to enhance the model's ability to integrate and utilize multi-level features effectively. Then, Spatial and Channel Reconstruction Detect Head (SC_Detect) based on spatial and channel preprocessing is designed to enhance the model's overall detection performance. Finally, the channel pruning technique is used to reduce the redundancy of the improved model, drastically reduce Parameters and Giga Floating Point Operations Per Second (GFLOPs), and improve the detection efficiency. The experimental results show that the mAP50 and mAP50:95 of the MRS-YOLO algorithm proposed in this paper are improved to 94.8% and 86.4%, respectively, which are 0.7 and 2.3 percentage points higher compared to the baseline, while Parameters and GFLOPs are reduced by 44.2% and 17.5%, respectively. It is demonstrated that the improved algorithm can be better applied to the task of foreign object detection in railroad transmission lines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。