基于YOLOv5的锂电池缺陷检测方法、系统、设备以及存储介质技术方案

本发明专利技术公开了一种基于YOLOv5的锂电池缺陷检测方法、系统、设备以及存储介质。该方法包括以下步骤：S1：获取原图像，将原图像的有效检测区域框选出来得到有效检测区域图像；S2：将有效检测区域图像缩放至统一尺寸，并将有效检测区域图像分割成适应模型输入的统一尺寸的多个小图像；S3：对分割后的多个小图像进行数据标注，并对在更大数据集上训练好的YOLOv5模型进行应用并完成训练；S4：训练完成后，使用训练得到的模型进行推理得到模型推理结果，并根据模型推理结果将多个小图像进行拼接，使其还原至有效检测区域图像大小，以得到锂电池缺陷检测的结果。本发明专利技术能够降低锂电池缺陷检测的误检率和漏检率，有效提高检测效率。有效提高检测效率。有效提高检测效率。

【技术实现步骤摘要】
基于YOLOv5的锂电池缺陷检测方法、系统、设备以及存储介质


[0001]本专利技术涉及工业自动化机器视觉
，尤其涉及一种基于YOLOv5的锂电池缺陷检测方法、系统、设备以及存储介质。

技术介绍

[0002]在生产加工过程中，因工艺限制或操作失误，锂电池表面难免会出现划痕、凹坑等缺陷。这些缺陷除了会降低产品质量外，还有可能造成安全隐患。因此，必须对锂电池的表面进行缺陷检测。在工业自动化机器视觉领域，目标检测算法逐渐取代了传统机器视觉在缺陷检测领域的地位。近年来，得益于算法通用性强和背景误检率低的优点，YOLO及其改进算法受到市场的青睐。而YOLOv5模型是YOLO系列算法中检测速度及精度表现良好的检测算法。目前市面上很多基于YOLOv5的算法都是对整张图像进行处理，这种处理的缺点有以下两点：一是锂电表面缺陷检测行业通常采集到的作为模型输入的样本图像尺寸较大，通用开源模型未能做很好的适配；二是在大尺度图像的输入上，划痕所占像素通常只占整张图像像素的很小一部分，一些小划痕不易检出或是检出的置信度不高。

技术实现思路

[0003]本专利技术要解决的技术问题在于，针对现有技术存在的至少一个缺陷，提供一种基于YOLOv5的锂电池缺陷检测方法、系统、设备以及存储介质。
[0004]本专利技术解决其技术问题所采用的技术方案是：构造一种基于YOLOv5的锂电池缺陷检测方法，包括以下步骤：
[0005]S1：获取原图像，将所述原图像的有效检测区域框选出来得到有效检测区域图像；
[0006]S2：将所述有效检测区域图像缩放至统一尺寸，并将所述有效检测区域图像分割成适应模型输入的统一尺寸的多个小图像；
[0007]S3：对分割后的所述多个小图像进行数据标注，并对在更大数据集上训练好的YOLOv5模型进行应用并完成训练；
[0008]S4：训练完成后，使用训练得到的模型进行推理得到模型推理结果，并根据所述模型推理结果将所述多个小图像进行拼接，使其还原至所述有效检测区域图像大小，以得到锂电池缺陷检测的结果。
[0009]优选地，在执行所述步骤S1之后还包括：
[0010]对所述有效检测区域图像进行平滑去噪处理。
[0011]优选地，相邻两个所述小图像的临边区域形成有12％
‑
18％的重叠面积。
[0012]优选地，在执行所述步骤S4之前还包括：
[0013]训练完成后，获得模型权重，使用测试集测试模型的准确度。
[0014]优选地，所述步骤S2中的将所述有效检测区域图像缩放至统一尺寸包括：将所述有效检测区域图像的宽和高等分，并使所述有效检测区域图像的长边尺寸设置到与YOLOv5模型的输入尺寸相近。
[0015]优选地，在执行所述步骤S4之后，还包括：
[0016]对拼接后的所述效检测区域图像进行非极大值抑制处理。
[0017]本专利技术还提供一种基于YOLOv5的锂电池缺陷检测系统，该系统包括：
[0018]有效检测区域图像获取模块，用于获取原图像，将所述原图像的有效检测区域框选出来得到有效检测区域图像；
[0019]分割模块，用于将所述有效检测区域图像缩放至统一尺寸，并将所述有效检测区域图像分割成适应模型输入的统一尺寸的多个小图像；
[0020]模型训练模块，用于对分割后的所述多个小图像进行数据标注，并对在更大数据集上训练好的YOLOv5模型进行应用并完成训练；
[0021]缺陷检测结果获取模块，用于训练完成后，使用训练得到的模型进行推理得到模型推理结果，并根据所述模型推理结果将所述多个小图像进行拼接，使其还原至所述有效检测区域图像大小，以得到锂电池缺陷检测的结果。
[0022]本专利技术还提供一种电子设备，其特征在于，包括处理器和存储有计算机可读指令的存储器，所述处理器被配置在执行所述计算机可读指令时，执行如上述任一项所述的方法。
[0023]本专利技术还提供一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现如上述任一项所述的方法。
[0024]通过实施本专利技术，具有以下有益效果：
[0025]本专利技术可以根据检测需求裁剪任意位置任意形状的检测区域，且将有效检测区域图像进行分割和拼接，能够降低锂电池缺陷检测的误检率和漏检率，有效提高检测效率。
附图说明
[0026]下面将结合附图及实施例对本专利技术作进一步说明，附图中：
[0027]图1为本专利技术一些实施例中锂电池密封钉表面需检测的划痕缺陷；
[0028]图2为本专利技术一种基于YOLOv5的锂电池缺陷检测方法的处理流程图；
[0029]图3为原图像中有效检测区域的框选图；
[0030]图4为对有效检测区域图像进行分割处理的示意图；
[0031]图5为本专利技术一种基于YOLOv5的锂电池缺陷检测系统的框图；
[0032]图6a为没有进行图像切割预处理的验证结果图；图6b为进行图像切割预处理后的验证结果图；
[0033]图7a为没有进行图像切割预处理的推理结果图；图7b为进行图像切割预处理后的推理结果图。
具体实施方式
[0034]为了对本专利技术的技术特征、目的和效果有更加清楚的理解，现对照附图详细说明本专利技术的具体实施方式。
[0035]需要说明的是，附图中所示的流程图仅是示例性说明，不是必须包括所有的内容和操作/步骤，也不是必须按所描述的顺序执行。例如，有的操作/步骤还可以分解，而有的操作/步骤可以合并或部分合并，因此实际执行的顺序有可能根据实际情况改变。
[0036]附图中所示的方框图仅仅是功能实体，不一定必须与物理上独立的实体相对应。即，可以采用软件形式来实现这些功能实体，或在一个或多个硬件模块或集成电路中实现这些功能实体，或在不同网络和/或处理器装置和/或微控制器装置中实现这些功能实体。
[0037]在工业自动化机器视觉领域，锂电池表面缺陷检测通常在现场传回的图像尺寸相对较大，划痕对比度较低，如果仅仅使用YOLOv5的通用模型来检测是远远不够的。因此，需要考虑将原始输入图像提前进行分割的预处理。如图1所示是常见的锂电池密封钉表面需要检测产品图中的缺陷。一般的处理流程是先将图像做标注，然后将标注后的图像文件和原图像一起导入模型进行模型训练，最后得到模型训练结果进行缺陷的预测。在检测阶段中，给予输入图像进行小目标检测，以输入图像2048
×
2048为例，yolov5中进行了5次下采样，最后得到的特征图大小分别是64
×
64、128
×
128与256
×
256，这3个中最大的256
×
256负责检测小目标，对应到2048
×
2048上，每个特征图的感受视野尺寸大小是8
×
8。
[0038]具体地，如图2所示，本专利技术公开了一种基于YOLOv5的锂电池缺陷检测方法，该方法包括以下步骤：...

【技术保护点】

【技术特征摘要】
1.一种基于YOLOv5的锂电池缺陷检测方法，其特征在于，包括以下步骤：S1：获取原图像，将所述原图像的有效检测区域框选出来得到有效检测区域图像；S2：将所述有效检测区域图像缩放至统一尺寸，并将所述有效检测区域图像分割成适应模型输入的统一尺寸的多个小图像；S3：对分割后的所述多个小图像进行数据标注，并对在更大数据集上训练好的YOLOv5模型进行应用并完成训练；S4：训练完成后，使用训练得到的模型进行推理得到模型推理结果，并根据所述模型推理结果将所述多个小图像进行拼接，使其还原至所述有效检测区域图像大小，以得到锂电池缺陷检测的结果。2.根据权利要求1所述的基于YOLOv5的锂电池缺陷检测方法，其特征在于，在执行所述步骤S1之后还包括：对所述有效检测区域图像进行平滑去噪处理。3.根据权利要求1所述的基于YOLOv5的锂电池缺陷检测方法，其特征在于，相邻两个所述小图像的临边区域形成有12％
‑
18％的重叠面积。4.根据权利要求1所述的基于YOLOv5的锂电池缺陷检测方法，其特征在于，在执行所述步骤S4之前还包括：训练完成后，获得模型权重，使用测试集测试模型的准确度。5.根据权利要求1所述的基于YOLOv5的锂电池缺陷检测方法，其特征在于，所述步骤S2中的将所述有效检测区域图像缩放至统一尺寸包括：将所述有效检测区域图像的宽和高等分，并使...

【专利技术属性】
技术研发人员：谢崇亮，麦浩晃，刘立峰，王典雄，
申请(专利权)人：昂视智能深圳有限公司，
类型：发明
国别省市：