[发明专利]用于多类别瑕疵实时检测的轻量化模型构建方法及装置

说明书

本发明公开了一种用于多类别瑕疵实时检测的轻量化模型构建方法及装置，方法包括如下步骤：获取含多类别瑕疵的目标物体图像，形成图像数据集；搭建初始的轻量化模型，并设置轻量化模型训练的超参数，其中，初始的轻量化模型包括轻量主干网络、颈部网络和头部检测网络，轻量主干网络用于提取融合位置信息的多尺度图像特征；根据图像数据集对初始的轻量化模型进行模型训练迭代，至收敛；构建得到用于多类别瑕疵实时检测的轻量化模型。本发明给出的轻量化模型提升了多类别缺陷产品的生产设备的工作效率和产品表面瑕疵的检测准确率。

技术领域

本发明属于图像识别技术领域，具体涉及用于多类别瑕疵实时检测的轻量化模型构建方法及装置。

背景技术

当前的一些产品在生产过程中，可能存在多种类别的瑕疵。如作为生活必需用品，卫生巾具有干净、舒适、功能强大的特点，已经成为了目前市面上必不可少的一类生活用品。在卫生巾的生产工艺过程中，需要将棉花、不织布、纸浆及吸水高分子材料等多种原料通过机器进行混合、整理、成型，并在机器传送过程中，完成指定芯体形状的内层压合以及两侧无纺布的切刀操作，最后使用包膜进行纵向和横向的封装。

在自动生产线的多工序产出过程中，由于原料梳理混乱、芯体分层不当以及部件错位等原因，易导致产品的不合格。因此，为保证卫生巾生产过程的可靠运行，设置产品在生产线上进行瑕疵检测与分类成为了一项十分有意义的研究。

如专利CN113514369A给出一种一次性卫生用品使用性能测试方法，包括以下步骤：1)选取至少八个重量在标准范围值内的产品放置于工作台上；2)通过自动量取加液装置依次对各个产品进行自动加液，并通过图像识别单元获取产品图像；3)通过获取的产品图像识别液体被完全吸收的状态，并自动计时吸收时间；4)将产品图像二值化处理，使得产品上的扩散图形形成斑点；5)提取斑点分布图数据形成虚拟图像；6)将虚拟图像与预先设置的尺寸图像比较，尺寸图像为呈矩阵分布的具有确定尺寸大小的测量单元格组成，通过识别覆盖于虚拟图像上的斑点的单元格个数以及位置。该方案主要解决了现有技术中一次性卫生用品的使用性能测量速度慢且测量误差大。

现有的卫生巾瑕疵检测方法主要是基于卫生巾产品图像的对比进行检查。通过与标准卫生巾产品的形状、花纹对比，剔除不合格的不良品。如以上现有技术的方法在检测种类较多时，存在准确度低、识别速度慢等问题，难以精确检测瑕疵位置。

因此，如何构建实时检测多类别瑕疵产品的轻量化模型，能够实现检测的高效、高精度是本领域技术人员亟待解决的问题。

发明内容

针对上述现有技术中存在的缺陷，本发明提供了用于多类别瑕疵实时检测的轻量化模型构建方法及装置，搭建包括轻量主干网络的初始的轻量化模型，并对初始的轻量化模型进行迭代训练，旨在从深度学习领域出发，构建针对多类别瑕疵实时检测的轻量化模型。以此，对多瑕疵类型进行识别分类，旨在提升产品生产设备的工作效率和多类别表面瑕疵产品的检测准确率，有助于对不良品进行瑕疵类别的整理与数据分析，从而进一步评估和改良产品的生产线。

第一方面，本发明提供一种用于多类别瑕疵实时检测的轻量化模型构建方法，包括如下步骤：

获取含多类别瑕疵的目标物体图像，形成图像数据集；

搭建初始的轻量化模型，并设置轻量化模型训练的超参数，其中，初始的轻量化模型包括轻量主干网络、颈部网络和头部检测网络，轻量主干网络用于提取融合位置信息的多尺度图像特征；

根据图像数据集对初始的轻量化模型进行模型训练迭代，至收敛；

构建得到用于多类别瑕疵实时检测的轻量化模型。

进一步的，获取含多类别瑕疵的目标物体图像，形成图像数据集，还包括：

对图像的瑕疵进行矩形框标记，并对应生成瑕疵信息，其中，瑕疵信息包括瑕疵类别、矩形框中心坐标及矩形框尺寸。