重庆理工大学学报（自然科学）

重庆理工大学学报（自然科学）

• •    

一种改进多尺度融合的电动汽车充电口识别方法

  

  1. 1.河北科技大学 信息科学与工程学院，石家庄 050018；2.中国人民解放军93507部队，石家庄 050200

PDF (PC)

371

赞

摘要： 针对无人自动充电桩工作过程中背景复杂导致电动汽车充电口识别精度低等问题，提出基于改进YOLOv5算法的一种电动汽车充电口识别方法。首先采用融入加权双向特征金字塔结构增强不同层级间的信息融合能力；其次引入 GhostNet 网络结构中的深度可分离卷积 GhostConv，替代原模型中的特征提取网络中的普通卷积层，减小了模型的计算开销；主干网络中使用SENet结构增加感受野信息，提高模型对充电口特征的提取能力；同时改进模型损失函数，引入EIoU损失函数代替原始CIoU损失函数，提高边界框回归精度。实验结果表明：改进后的模型在自制的多样化电动汽车充电口数据集上相较于原始YOLOv5算法在模型体积上减小了 6.94 MB，检测精度提升到94.75%。同时与目前主流的检测算法对比在检测精度与检测速度上也具有一定的优越性，适用于复杂背景环境下的电动汽车充电口的目标检测。

关键词: 图像处理, 目标检测, 电动汽车充电口, 注意力机制, 多尺度融合, YOLOv5