基于MobileNet的移動端列車圖像故障檢測算法

打開文本圖片集

摘  要：針對列車故障檢測效率低的問題，提出一種基于MobielNet的移動端列車圖像故障檢測算法。首先，在MobileNet中引入注意力卷積塊和Ghost模塊，用以提升網(wǎng)絡的學習能力。其次，使用殘差聚合網(wǎng)絡獲取多層次的特征圖。最后，將該模型移植到移動端設(shè)備上完成列車故障檢測任務。實驗結(jié)果表明，該算法的平均精度均值達到了85.35%，與YOLOv3-Tiny、YOLOv4-Tiny、YOLOX、YOLOv5相比，mAP分別提高了8.83%、5.49%、7.89%、5.31%，并且FED擁有更低檢測延遲。（剩余8186字）