首页|基于多尺度特征增强的列车摘钩关键部件检测

基于多尺度特征增强的列车摘钩关键部件检测

扫码查看
原文链接
  • NETL
  • NSTL
  • 万方数据
  • 维普
针对列车摘钩关键部件风管和折角塞门的小尺寸目标检测问题,使用多尺度特征增强改进YOLOv5s检测算法.首先在主干网络增加一个多分支卷积模块C2f输出一个大尺寸特征图;然后在颈部结构采用残差特征增强方式RFA提升特征金字塔的多尺度特征融合表征能力;最后检测头输出 4 种尺度的检测特征图,其中新增特征图的尺寸最大,用于增强小尺寸目标的检测能力.实验结果表明,改进 YOLOv5s在平均检测精度mAP 上提升了1.4%,优于SSD,提升了小尺寸风管和折角塞门的检测能力.

潘培玉、刘宏州、孙欣、李瑞、石磊

展开 >

陕西榆横铁路有限责任公司 陕西榆林 719000

四川国软科技集团有限公司 四川成都 610031

铁路货运 列车摘钩部件 目标检测 多尺度特征 残差特征增强

四川省经济和信息化厅-中国制造2025四川行动资金项目陕西煤业化工集团有限责任公司2021年度板块级科研项目

CJXDZ-Y2021-0032021SMHKJ-BK-J-52

2024

10.3969/j.issn.1672-9528.2024.01.026

信息技术与信息化
山东电子学会

信息技术与信息化

影响因子：0.29
ISSN：1672-9528
年,卷(期)：2024.(1)
  • 18