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

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

扫码查看
针对无人自动充电桩工作过程中背景复杂导致电动汽车充电口识别精度低等问题,提出基于改进YOLOv5算法的一种电动汽车充电口识别方法.采用融入加权双向特征金字塔结构增强不同层级间的信息融合能力;引入GhostNet网络结构中的深度可分离卷积 GhostConv,替代原模型中的特征提取网络中的普通卷积层,减小了模型的计算开销;主干网络中使用SENet结构增加感受野信息,提高模型对充电口特征的提取能力;同时改进模型损失函数,引入EIoU损失函数代替原始CIoU损失函数,提高边界框回归精度.实验结果表明:改进后的模型在自制的多样化电动汽车充电口数据集上相较于原始YOLOv5 算法在模型体积上减小了 6.94 MB,检测精度提升到94.75%.同时与目前主流的检测算法对比,检测精度与检测速度也具有一定的优越性,适用于复杂背景环境下的电动汽车充电口的目标检测.
An improved multi-scale fusion identification method for electric vehicle charging ports
To address the low accuracy in electric vehicle charging port recognition due to the complex background during the operation of unmanned automatic charging stations,this paper proposes a charging port recognition method for electric vehicles based on the improved YOLOv5 algorithm.First,the method incorporates a weighted bidirectional feature pyramid structure to enhance information fusion capabilities between different levels.Second,it introduces GhostConv,a depthwise separable convolution from the GhostNet network structure,replacing the ordinary convolution layers in the original feature extraction network,reducing the computational overhead of the model.The main network employs the SENet structure to increase the receptive field information,enhancing the model's ability to extract charging port features.Meanwhile,the loss function of the model is improved by introducing the EIoU loss function to replace the original CIoU loss function,enhancing the accuracy of bounding box regression.Our experimental results demonstrate the improved model,compared to the original YOLOv5 algorithm,reduces the model size by 6.94 MB and achieves a detection accuracy of 94.75%on a self-made,diverse dataset of electric vehicle charging ports.Furthermore,compared to the mainstream detection algorithms,it delivers superior detection accuracy and speed,making it suitable for target detection of electric vehicle charging ports in complex background environments.

picture processingobject detectionelectric vehicle charging portattention mechanismmultiscale fusionYOLOv5

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

展开 >

河北科技大学 信息科学与工程学院,石家庄 050018

中国人民解放军93507部队,石家庄 050200

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

2024

重庆理工大学学报
重庆理工大学

重庆理工大学学报

CSTPCD北大核心
影响因子:0.567
ISSN:1674-8425
年,卷(期):2024.38(13)