一种基于改进YOLOv4的锂电池缺陷检测方法

摘要

本发明涉及一种基于改进YOLOv4的锂电池缺陷检测方法，包括步骤：1)获取锂电池表面缺陷图像；2)对获取的锂电池表面缺陷图像进行缺陷类型和缺陷位置标注，生成数据集，对数据集进行预处理，并将处理后的数据集划分为训练集和测试集；3)改进YOLOv4网络模型的主干特征提取网络，加入空洞卷积扩大感受野，获取优化后的改进YOLOv4网络模型，基于训练集对优化后的改进YOLOv4网络模型进行训练；4)利用训练后的改进YOLOv4网络模型对锂电池表面缺陷进行测试。与现有技术相比，本发明具有检测精度高、降低漏检率等优点。