基于改进YOLOv5的锂电池极片缺陷检测方法
Defect detection method of lithium battery electrode based on improved YOLOv5
冉庆东 1郑力新2
作者信息
- 1. 华侨大学信息科学与工程学院,福建厦门 361021
- 2. 华侨大学工学院,福建泉州 362021
- 折叠
摘要
针对同时存在多种小目标、大长宽比目标缺陷的锂电池极片复杂表面,基于可变形卷积和YOLOv5 提出DDCNet-YOLO算法模型.在主干网络部分构建出可变形下采样卷积主干网络(DDCNet),在特征融合部分引入上下文增强模块(CAM),并使用构造的可变形卷积块(DCB)替换C3 模块,在检测头部分设计带有注意力机制的解耦头 AD-Head.提出 RIoU方法优化不同长宽比目标的损失计算.实验表明,DDCNet-YOLO模型相较于YOLOv5s及YOLOv5m模型在mAP50 上分别提高了 6.2 个百分点和 3.7 个百分点.仅通过DDCNet和注意力机制解耦头构建了DDCNet-YOLOs轻量化模型,与YOLOv5s模型相比,参数量减少 7.2 个百分点,mAP50∶95 提升 8.9 个百分点.对 2 种模型通过C++的方式进行了部署.本研究所提出的 2 种算法模型分别侧重于精度和轻量化,都能够在满足一定实际检测速度的条件下,达到较高的检测精度.
Abstract
The DDCNet-YOLO algorithm model was proposed based on the deformable convolution and YOLOv5,aiming at the complex lithium battery electrode surface with multiple small object defects and large aspect ratio object defects at the same time.The deformable downsampling convolution network(DDCNet)was constructed in the backbone.The context augmentation module(CAM)was introduced in the feature fusion part and the deformable convolution block(DCB)was used to replace the C3 module.AD-Head,a decoupling head with an attention mechanism,was designed in the head part.The RIoU method was proposed to optimize the loss calculation for different aspect ratio objects.Experiments showed that the DDCNet-YOLO model improved the mAP50 by 6.2 percentage points compared to YOLOv5s model and by 3.7 percentage points compared to YOLOv5m model.The lightweight model DDCNet-YOLOs,constructed by DDCNet and a decoupling head with an attention mechanism.The DDCNet-YOLOs improved the mAP50:95 by 8.9 percentage points and reduced the number of parameters by 7.2 percentage points,compared with the YOLOv5s model.In addition,both models were deployed based on the C++.The two algorithmic models focus on accuracy and speed respectively,but both can achieve high accuracy under the condition of meeting the actual detection speed requirement.
关键词
极片缺陷/可变形卷积/小目标/大长宽比目标/YOLOv5Key words
electrode defect/deformable convolution/small object/large aspect ratio object/YOLOv5引用本文复制引用
基金项目
福建省科技计划资助项目(2020Y0039)
出版年
2024
NETL
NSTL
万方数据