[发明专利]基于知识蒸馏的高速瓦楞纸箱印刷缺陷检测方法、系统及存储介质

说明书

本发明公开了一种基于知识蒸馏的高速瓦楞纸箱印刷缺陷检测方法、系统、设备及存储介质，方法包括：获取瓦楞纸箱印刷缺的被测目标图像，基于被测目标图像构建样本数据集，构建知识蒸馏网络模型，所述知识蒸馏网络模型包括教师网络模型和学生网络模型；对所述教师网络模型和所述学生网络模型分别进行训练，将样本数据集输入到主干网络中获取样本数据集的特征图，在特征图的每一点上面进行回归操作；利用训练好的知识蒸馏网络模型对高速瓦楞纸箱印刷缺陷进行检测。本发明利用知识蒸馏网络提高了高速瓦楞纸箱印刷缺陷检测的精度。

技术领域

本发明属于缺陷检测的技术领域，具体涉及一种基于知识蒸馏的高速瓦楞纸箱印刷缺陷检测方法、系统及存储介质。

背景技术

随着近几年深度学习的快速发展，许多深度学习算法引用到各个行业。在工业界，质量检测是工业自动化中重要的一环，基于深度学习的缺陷检测技术已经应用到许多工业场景代替人工肉眼检测，包括电子、包装和印刷等行业。

目前大多数基于深度学习的方法都需要在大规模数据集上训练模型，以实现特定的工业智能应用。大量的计算能力和内存资源消耗限制了基于深度学习的方法在涉及移动或嵌入式设备等边缘设备部署的工业智能应用中的推广和进步。而知识蒸馏深度学习方法可以简单地在不同的深度模型上实现，并且可以通过知识蒸馏轻松地实现不同深度模型的模型压缩，同时可以很好地改善深度模型的性能。因此，基于知识蒸馏的高速瓦楞纸箱印刷缺陷检测方法显得格外重要，一定程度解决了深度学习在工业图像缺陷检测的局限性问题，而且有着非常广阔的市场应用前景。

发明内容

本发明的主要目的在于克服现有技术的缺点与不足，提供一种基于知识蒸馏的高速瓦楞纸箱印刷缺陷检测方法、系统及存储介质，传统深度学习方法在在工业图像缺陷检测通常需要大规模数据集训练，得到的复杂和大规模参数网络模型部署到边缘设备中需要消耗大量内存资源和限制工业检测速度；本发明通过训练得到一个轻量级深层模型，能够在不损失缺陷检测准确性的同时保持较高的实时性能。

为了达到上述目的，本发明采用以下技术方案：

本发明提供了一种基于知识蒸馏的高速瓦楞纸箱印刷缺陷检测方法，包括下述步骤：

获取瓦楞纸箱印刷缺的被测目标图像，基于被测目标图像构建样本数据集，

构建知识蒸馏网络模型，所述知识蒸馏网络模型包括教师网络模型和学生网络模型；

对所述教师网络模型和所述学生网络模型分别进行训练，所述教师网络模型的训练具体为：将样本数据集输入到主干网络中获取样本数据集的特征图，在特征图的每一点上面进行回归操作，进行网络训练获取教师网络模型；所述学生网络模型的训练具体为：通过训练后的教师网络模型进行诱导训练，将低分辨率图像输入到主干网络中获取输入图像的特征图，在特征图的每一点上面进行回归操作，将教师网络模型的预测输出作为标签，将学生网络模型的预测输出作为软标签，将真实标签作为硬标签，并计算学生网络模型的损失，更新网络参数，实际应用仅使用学生网络模型；

利用训练好的知识蒸馏网络模型对高速瓦楞纸箱印刷缺陷进行检测。

作为优选的技术方案，所述教师网络模型的训练过程表示如下：

G_IT＝F_T(I_h,heads₁)

其中，F_T(·)为一个神经网络函数，表示教师网络模型，heads₁为网络预测的结果参数，I_h表示训练教师网络模型输入图像。

作为优选的技术方案，所述低分辨率图像通过下述方式获得：