重庆理工大学学报(自然科学) ›› 2024, Vol. 38 ›› Issue (7): 118-126.

• 信息计算机 • 上一篇    下一篇

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

赵晓东,刘瑞庆,王 向,温士涛   

  1. 1.河北科技大学信息科学与工程学院,石家庄 050018;2.中国人民解放军93507部队,石家庄 050200
  • 发布日期:2024-08-13
  • 作者简介:赵晓东,男,博士,研究员,主要从事电子、智能电器及物联网研究,E-mail:zhaoxiaodong@hebust.edu.en;通信作者王向,女,博士,副教授,主要从事智能信息处理等研究,E-mail:wangxiang@hebust.edu.en。

  • Published:2024-08-13

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

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

中图分类号: 

  • TP391.41