基于MobileNet的移动端列车图像故障检测算法

作者：周鹏; 张龙信

来源：现代信息科技, 2023, 7(06): 46-53.

DOI：10.19850/j.cnki.2096-4706.2023.06.012

• 列车故障检测
• 目标识别
• MobileNet
• 移动设备
• 注意力机制

针对列车故障检测效率低的问题，提出一种基于MobielNet的移动端列车图像故障检测算法。首先，在MobileNet中引入注意力卷积块和Ghost模块，用以提升网络的学习能力。其次，使用残差聚合网络获取多层次的特征图。最后，将该模型移植到移动端设备上完成列车故障检测任务。实验结果表明，该算法的平均精度均值达到了85.35%，与YOLOv3-Tiny、YOLOv4-Tiny、YOLOX、YOLOv5相比，mAP分别提高了8.83%、5.49%、7.89%、5.31%，并且FED拥有更低检测延迟。

• 出版日期2023
• 单位湖南工业大学