文档名：基于RP图像注意融合网络的列车轮对踏面缺陷识别
摘要：轮对踏面缺陷识别是开展列车轮对检修维护的重要前提.然而,轮对踏面缺陷样本存在的数量少、类别不均衡现象,使得传统深度学习方法识别精度有限,难以满足列车智能运维发展的需求.提出一种基于R-P图像注意融合网络(RP-AFN)的列车轮对踏面缺陷识别方法.首先,将RGB图像及其泊松编码(Poissoncoding,POS)形式同时引入Mobilenetv2网络之中,以通过丰富输入信息的方式来缓解小样本问题;其次,将挤压-激励模块(SEBlock)和相似性约束引入模型中,以提取具有模态间交互性信息的判别注意力特征,并利用多层感知机(MLP)对其进行融合和推断;最后,基于焦点损失和相似性损失设计一种联合损失函数,使其能够通过惩罚大类别样本以缓解类别不平衡问题.实验结果表明:所提融合方法实现了85.7％的轮对踏面缺陷识别率,且其各项指标以约2.5％～20％的优势优于对比方法.在不同输入大小和主干网络的条件下,以224×224为输入大小的Mobilenetv2模型所对应的模型性能最佳.在不同特征融合策略的条件下,特征拼接方法的各项指标以约1％～3.5％的优势优于特征加法、特征外积和动态加权等方法.在不同调剂因子μ的设置中,μ=0.7的情况下,模型各项指标性能达到综合最优.在主干网络相同的条件下,融合方法能够在不显著增加模型参数的情况下,其各项指标优于单模态方法约1％～6.3％.在执行消融实验的情况下,输入模态、损失函数和模块的消融都会给模型带来负面影响,验证了模型设计的合理性.RP-AFN模型能够有效地提高轮对踏面缺陷识别性能,可在一定程度上解决小样本和类别不均衡条件下的踏面缺陷识别问题.
作者：杨能普 周苗 王文昆 张娟 孙永奎 Author：YANGNengpu ZHOUMiao WANGWenkun ZHANGJuan SUNYongkui
作者单位：湖南工业大学轨道交通学院,湖南株洲412007株洲时代电子技术有限公司,湖南株洲412007中铁特货大件运输有限公司,北京100070北京交通大学轨道交通运行控制系统国家工程研究中心,北京100044
刊名：铁道科学与工程学报 ISTICPKU
Journal：JournalofRailwayScienceandEngineering
年，卷(期)：2023, 20(12)
分类号：U279
关键词：轮对踏面缺陷识别 图像融合 深度学习 注意力机制
Keywords：wheelsettreaddefectrecognition imagefusion deeplearning attentionmechanism
机标分类号：TP391TP193TP751
在线出版日期：2024年1月24日
基金项目：国家重点研发计划，湖南省自然科学基金资助项目基于R-P图像注意融合网络的列车轮对踏面缺陷识别[
期刊论文] 铁道科学与工程学报--2023, 20(12)杨能普 周苗 王文昆 张娟 孙永奎轮对踏面缺陷识别是开展列车轮对检修维护的重要前提.然而,轮对踏面缺陷样本存在的数量少、类别不均衡现象,使得传统深度学习方法识别精度有限,难以满足列车智能运维发展的需求.提出一种基于R-P图像注意融合网络(RP-AFN)...参考文献和引证文献
参考文献
引证文献
本文读者也读过
相似文献
相关博文
基于R-P图像注意融合网络的列车轮对踏面缺陷识别 Wheelset tread defect recognition based on R-P image attention fusion network
基于R-P图像注意融合网络的列车轮对踏面缺陷识别.pdf
- 文件大小:
- 18.43 MB
- 下载次数:
- 60
-
高速下载
|
|
