[发明专利]模型确定、图像识别与工业质检方法、设备及存储介质

说明书

本申请实施例提供一种模型确定、图像识别与工业质检方法、设备及存储介质。其中，方法包括如下的步骤：确定训练过的第一图像识别模型；第一图像识别模型可识别多个预设前景类别；根据第一图像识别模型，确定多个不同的目标图像识别模型；多个不同的目标图像识别模型可识别的预设前景类别不同；多个预设前景类别中包括多个不同的目标图像识别模型可识别的预设前景类别；根据训练样本，分别对多个不同的目标图像识别模型进行训练；其中，多个不同的目标图像识别模型用于对待识别图像进行识别。本申请实施例提供的技术方案能够提高模型整体识别性能，尤其在工业质检领域，可提高瑕疵产品的瑕疵类别的识别准确率。

技术领域

本申请涉及视觉技术领域，尤其涉及一种模型确定、图像识别与工业质检方法、设备及存储介质。

背景技术

目前，工业智能化是未来的发展趋势。工业质检是工业生产中最重要的环节之一，也是工业转型升级的重要突破口。传统工业质检依靠人力，不仅效率低、出错率高，而且人力成本高，人员易流失。

现有技术中存在一些机器视觉质检方案，来代替人工质检，实现了产品缺陷或瑕疵的识别及分类。但是，使用现有的机器视觉质检方案进行质检，存在识别准确率较低的问题。

发明内容

鉴于上述问题，提出了本申请以提供一种解决上述问题或至少部分地解决上述问题的模型确定、图像识别与工业质检、设备及存储介质。

于是，在本申请的一个实施例中，提供了一种模型确定方法。该方法包括：

确定训练过的第一图像识别模型；所述第一图像识别模型可识别多个预设前景类别；

根据所述第一图像识别模型，确定多个不同的目标图像识别模型；所述多个不同的目标图像识别模型可识别的预设前景类别不同；所述多个预设前景类别中包括所述多个不同的目标图像识别模型可识别的预设前景类别；

根据训练样本，分别对所述多个不同的目标图像识别模型进行训练；

其中，所述多个不同的目标图像识别模型用于对待识别图像进行识别。

在本申请的又一实施例中，提供了一种图像识别方法。该方法，包括：

获取待识别图像；

将所述待识别图像分别输入至训练过的多个不同的目标图像识别模型，以得到多个识别结果；所述多个不同的目标图像识别模型可识别的预设前景类别不同；

根据所述多个识别结果，确定所述待识别图像的目标识别结果；

其中，所述多个不同的目标图像识别模型的确定过程包括：

确定训练过的第一图像识别模型；所述第一图像识别模型可识别多个预设前景类别；所述多个预设前景类别中包括所述多个不同的目标图像识别模型可识别的预设前景类别；

根据所述第一图像识别模型，确定所述多个不同的目标图像识别模型；

根据训练样本，分别对所述多个不同的目标图像识别模型进行训练。

在本申请的又一实施例中，提供了一种工业质检方法。该工业质检方法，包括：

获取待识别产品图像；

将所述待识别产品图像分别输入至训练过的多个不同的目标瑕疵识别模型，以得到多个识别结果；所述多个不同的目标瑕疵识别模型可识别的预设瑕疵类别不同；

根据所述多个识别结果，确定所述待识别产品图像的目标识别结果；

其中，所述多个不同的目标瑕疵识别模型的确定过程包括：

确定训练过的第一瑕疵识别模型；所述第一瑕疵识别模型可识别多个预设瑕疵类别；所述多个预设瑕疵类别中包括所述多个不同的目标瑕疵识别模型可识别的预设瑕疵类别；