基于特征降冗余的Vanilla-YOLOv8铁路异物侵限检测方法
杜开华
许贵阳
白堂博
北京建筑大学机电与车辆工程学院,北京 100044;北京建筑大学城市轨道交通车辆服役性能保障北京市重点实验室,北京 100044
摘要:铁路异物侵限在线监测技术是铁路运行安全和旅客生命财产安全的重要保障.针对当前异物检测算法存在的遮挡目标检测、小目标检测不全不准等问题,提出一种基于YOLOv8的铁路异物检测算法Vanilla-YOLOv8.首先,结合VanillaNet中减少网络深度、短接分支以及通过改变深度训练策略和动态调节激活函数状态增强非线性能力的思想,改善YOLOv8模型中由于网络层数过多、短接分支过盛带来的模型退化、耗时以及深层小目标特征消失等问题,提高模型的特征提取能力和检测速度;然后,利用改进的局部卷积减少冗余特征的出现,确保特征能被充分利用;最后,在网络主干部分融入压缩激励(Squeeze-and-Excitation,SE)注意力机制,提高网络中关键特征的权重,加强对遮挡目标和小目标的特征表征能力和检测能力.实验结果表明:Vanilla-YOLOv8 算法的平均精度均值达到 98.7%,参数量下降 61.39%,识别速度达到 125 帧每秒(Frames Per Second,FPS),速度和检测精度较传统的图像处理技术有较大提升.研究结果可以为在线实时监测提供参考.
关键词:异物侵限检测YOLOv8特征冗余VanillaNet
分类号:U216.3(铁路线路工程)
论文发表日期:2024-10-28
在线出版日期:2025-08-15(本平台首次上网日期,不代表文献的发表时间)
页数:10( 49-58 )
英文信息展开
北京交通大学学报

北京交通大学学报

CSTPCD北大核心CSCD
ISSN:1673-0291
年,卷(期):2024,48(5)
所属栏目:交通领域目标检测算法