CN109829483A - 缺陷识别模型训练方法、装置、计算机设备和存储介质 - Google Patents

Abstract

本申请涉及一种缺陷识别模型训练方法、装置、计算机设备和存储介质。所述方法包括：获取样本图像训练数据；获取与所述样本图像训练数据对应的辅助样本图像数据；以所述样本图像训练数据和所述辅助样本图像数据，构建训练样本集；根据所述训练样本集对初始模型进行训练，得到缺陷识别模型。在训练时，通过输入与样本图像训练数据对应的辅助样本图像数据，无需额外对样本图像训练数据提取特征数据，减少了模型训练时提取特征数据所耗费的时间，提高了模型的训练效率。

Description

缺陷识别模型训练方法、装置、计算机设备和存储介质

技术领域

本申请涉及计算机技术领域，特别是涉及一种缺陷识别模型训练方法、装置、计算机设备和存储介质。

背景技术

随着计算机技术的发展，出现了机器学习技术，通过建立模型、提供训练数据进行模型训练，利用训练后的模型对未知数据进行预测。机器学习可以广泛地应用于品质检测领域，建立的缺陷识别模型对包含缺陷的产品的图像进行学习，并通过产品的图像进行产品缺陷检测。产品可以为各种类型；产品的材质可以包括各种类型的材质，比如金属、塑料和玻璃等；缺陷的种类可以包括各种类型，比如缺少、多余、气泡、划痕和形状不符合规则等。

然而，传统的机器学习技术中，在对缺陷识别模型进行训练时需要输入大量的训练数据样本，缺陷识别模型又需要从训练样本数据中提取大量的特征数据，导致缺陷识别模型的训练时间较长，效率较低。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高模型训练效率的缺陷识别模型训练方法、装置、计算机设备和存储介质。

一种缺陷识别模型训练方法，所述方法包括：

获取样本图像训练数据；

获取与所述样本图像训练数据对应的辅助样本图像数据；

以所述样本图像训练数据和所述辅助样本图像数据，构建训练样本集；

根据所述训练样本集对初始模型进行训练，得到缺陷识别模型。

一种缺陷识别模型训练装置，所述装置包括：

数据获取模块，用于获取样本图像训练数据；

辅助获取模块，用于获取与所述样本图像训练数据对应的辅助样本图像数据；

样本集构建模块，用于以所述样本图像训练数据和所述辅助样本图像数据，构建训练样本集；

模型得到模块，用于根据所述训练样本集对初始模型进行训练，得到缺陷识别模型。

一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现以下步骤：

获取样本图像训练数据；

获取与所述样本图像训练数据对应的辅助样本图像数据；

以所述样本图像训练数据和所述辅助样本图像数据，构建训练样本集；

根据所述训练样本集对初始模型进行训练，得到缺陷识别模型。

一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现以下步骤：

获取样本图像训练数据；

获取与所述样本图像训练数据对应的辅助样本图像数据；

以所述样本图像训练数据和所述辅助样本图像数据，构建训练样本集；

根据所述训练样本集对初始模型进行训练，得到缺陷识别模型。

上述缺陷识别模型训练方法、装置、计算机设备和存储介质，获取样本图像训练数据以及与样本图像训练数据对应的辅助样本图像数据，无需额外对样本图像训练数据提取特征数据，以样本图像训练数据和辅助样本图像数据,构建训练样本集，再根据训练样本集对初始模型进行训练,得到缺陷识别模型。在训练时，通过输入与样本图像训练数据对应的辅助样本图像数据，减少了模型训练时提取特征数据所耗费的时间，提高了模型的训练效率。

附图说明

图1为一个实施例中缺陷识别模型训练方法的应用环境图；

图2为一个实施例中缺陷识别模型训练方法的流程示意图；

图3为一个实施例中获取辅助样本图像数据的步骤的流程示意图；

图4为一个实施例中构建训练样本集的步骤的流程示意图；

图5为一个实施例中得到缺陷识别模型的步骤的流程示意图；

图6为一个实施例中得到缺陷预判模型的步骤的流程示意图；

图7为一个实施例中样本图像训练数据中样本图像的示意图；

图8为一个实施例中预处理图像数据中预处理图像的示意图；

图9为一个实施例中无缺产品图像数据中无缺产品图像的示意图；

图10为一个实施例中缺陷标注数据的示意图；

图11为一个实施例中使用缺陷识别模型的步骤的流程示意图；

图12为一个实施例中缺陷识别结果的示意图；

图13为一个实施例中缺陷识别模型训练的示意图；

图14为一个实施例中缺陷识别模型训练装置的结构框图；

图15为一个实施例中计算机设备的内部结构图。

具体实施方式

为了使本申请的目的、技术方案及优点更加清楚明白，以下结合附图及实施例，对本申请进行进一步详细说明。应当理解，此处描述的具体实施例仅仅用以解释本申请，并不用于限定本申请。

本申请提供的缺陷识别模型训练方法，可以应用于如图1所示的应用环境中，应用环境中可以包括终端102和服务器104，终端102通过网络与服务器104进行通信。该方法既可以应用在终端102，也可以应用于服务器104。其中，终端102可以但不限于是各种工业计算机、个人计算机、笔记本电脑、智能手机、平板电脑。服务器104可以用独立的服务器或者是多个服务器组成的服务器集群来实现。

在一个实施例中，如图2所示，提供了一种缺陷识别模型训练方法，本实施例以该方法应用于图1中的终端为例进行说明，包括以下步骤：

步骤202，获取样本图像训练数据。

其中，样本图像训练数据是对存在品质缺陷的产品进行图像采集得到样本图像后，对样本图像添加缺陷标注数据得到的样本数据，用于对初始模型进行训练。

具体地，终端获取用户触发的模型训练指令，对模型训练指令进行解析，得到样本图像训练数据的存储地址，终端访问存储地址对应的存储空间，从访问的存储空间中提取存储的样本图像训练数据。

在一个实施例中，终端根据模型训练指令读取预先配置的模型训练脚本，从模型训练脚本中提取样本图像数据的存储地址，根据提取到的存储地址从存储空间中读取样本图像训练数据。

步骤204，获取与样本图像训练数据对应的辅助样本图像数据。

其中，辅助样本图像数据是与样本图像训练数据相关的图像数据，用于辅助初始模型进行训练。

具体地，终端对获取到的模型训练指令进行解析，得到与样本图像数据对应的辅助样本图像数据类型，以及辅助样本图像数据的存储地址。终端根据辅助样本图像数据对应的存储地址，从存储空间中提取辅助样本图像数据。

在一个实施例中，终端根据模型训练指令生成数据获取请求，通过网络将数据获取请求发送至服务器。服务器接收数据获取请求，根据数据获取请求从数据库中提取样本图像训练数据和辅助样本图像数据，通过网络将提取到的样本图像训练数据和辅助样本图像数据发送至终端。

步骤206，以样本图像训练数据和辅助样本图像数据，构建训练样本集。

其中，训练样本集是样本图像训练数据和辅助样本图像数据的组合。

具体地，终端获取到样本图像训练数据及辅助样本图像数据后，触发训练样本集构建指令，根据训练样本集构建指令，对样本图像训练数据和辅助样本图像数据进行组合，得到训练样本集。

步骤208，根据训练样本集对初始模型进行训练，得到缺陷识别模型。

其中，初始模型是预先建立的、未经参数调整的模型。缺陷识别模型是初始模型经过训练后，最终得到的模型。

具体地，终端构建训练样本集完毕后，将训练样本集输入初始模型。初始模型依据输入的训练样本集进行训练，调整模型参数，使初始模型输出的缺陷预测结果不断接近缺陷标注数据。当训练结束后，终端得到缺陷识别模型。初始模型的种类在此不作限定，可以是神经网络模型、点云分类模型、支持向量机和逻辑回归模型中的至少一种。辅助样本图像数据种类的数量没有限定。比如，终端可以只获取一种辅助样本图像数据，构建一组训练样本集，根据训练样本集对一个初始模型进行训练得到缺陷识别模型。

本实施例中，通过获取样本图像训练数据以及与样本图像训练数据对应的辅助样本图像数据，无需额外对样本图像训练数据提取特征数据，以样本图像训练数据和辅助样本图像数据,构建训练样本集，根据训练样本集对初始模型进行训练,得到缺陷识别模型。在训练时，通过输入与样本图像训练数据对应的辅助样本图像数据，减少了模型训练时提取特征数据所耗费的时间，提高了模型的训练效率。

如图3所示，在一个实施例中，步骤204具体还包括获取辅助样本图像数据的步骤，该步骤具体包括如下步骤：

步骤302，确定样本图像训练数据对应的辅助数据类型。

具体地，终端提取样本图像训练数据标识，从预先存储的模型训练脚本中查询样本图像训练数据标识，确定与样本图像训练数据标识对应的辅助数据类型。模型训练脚本是对模型进行训练所用的格式文件，模型训练脚本中包括控制终端进行一系列操作的数据。辅助数据类型可以包括预处理图像数据、无缺产品图像数据和光照变换图像数据中的至少一种。

步骤304，获取各辅助数据类型分别对应的样本数据地址。

其中，样本数据地址是各辅助数据类型在存储空间中的存储地址。

具体地，终端根据模型训练脚本确定辅助数据类型后，在模型训练脚本中提取与确定的各辅助数据类型分别对应的样本数据地址。

步骤306，根据样本数据地址分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据。

其中，预处理图像数据是对样本图像进行视觉方法处理得到的预处理图像，添加标注数据后得到的图像数据。视觉方法的处理可以包括边缘提取、连通域和RGB转HSB中的至少一种。RGB表示一种色彩模式，通过对红(R)、绿(G)、蓝(B)三个颜色通道的变化以及它们相互之间的叠加得到各式各样的颜色；HSB表示一种颜色模式，H(hues)表示色相，S(saturation)表示饱和度，B(brightness)表示亮度，HSB模式对应的媒介是人眼。

无缺产品图像数据是对品质完好的产品进行图像采集得到的无缺产品图像，添加标注数据后得到的图像数据。

光照变换图像数据是对存在品质缺陷的产品，在特定颜色光照下进行图像采集得到的光照变换图像，添加标注数据后得到的图像数据。在对不同颜色的产品进行图像采集时，可以采用不同波长的光进行照射，采集产品在特定光照下的图像。比如，当产品为黑色时，可以使用蓝光或者绿光进行照射。在进行图像采集时，可以由人工确定产品缺陷类型，依据产品缺陷类型采用不同波长的光进行照射。比如，当产品表面存在细小划痕时，可以采用短波长的光进行照射，例如蓝光。

具体地，终端从模型训练脚本中获取到预处理图像数据、无缺产品图像数据和光照变换图像数据分别对应的样本数据地址后，分别访问各样本数据地址对应的存储空间，从访问的存储空间中，分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据。

在一个实施例中，在辅助样本图像数据的准备阶段，可以先对样本图像进行复制，对复制得到的样本图像进行传统视觉方法的处理，得到预处理图像；也可以从样本图像中选取一部分进行传统视觉方法的处理，得到预处理图像。

步骤308，以预处理图像数据、无缺产品图像数据和光照变换图像数据，作为各种辅助样本图像数据,

具体地，终端从存储空间中提取到预处理图像数据、无缺产品图像数据和光照变换图像数据后，分别将预处理图像数据、无缺产品图像数据和光照变换图像数据标记为辅助样本图像数据。

本实施例中，通过确定样本图像训练数据对应的辅助数据类型，获取各辅助数据类型分别对应的样本数据地址，根据样本数据地址分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据，再分别将提取到的预处理图像数据、无缺产品图像数据和光照变换图像数据作为辅助样本图像数据，提高了获取辅助样本图像数据的准确性。

如图4所示，在一个实施例中，步骤206具体还包括构建训练样本集的步骤，该步骤具体包括如下步骤：

步骤402，以样本图像训练数据作为第一组训练样本集。

具体地，终端获取到各种辅助样本图像数据后，触发训练样本集构建指令，根据训练样本集构建指令构建训练样本集。首先，终端单独以样本图像训练数据构建训练样本集，得到第一组训练样本集。

步骤404，将样本图像训练数据，分别与各种辅助样本图像数据中的预处理图像数据、无缺产品图像数据和光照变换图像数据进行组合，得到第二组训练样本集、第三组训练样本集和第四组训练样本集。

具体地，终端分别将样本图像训练数据分别于各种辅助样本图像数据进行组合，对于每种组合方式，终端分别得到第二组训练样本集、第三组训练样本集和第四组训练样本集。其中，第二组训练样本集可以是样本图像数据和预处理图像数据的组合，第三组训练样本集可以是样本图像数据和无缺产品图像数据的组合，第四组训练样本集可以是样本图像数据和光照变换图像数据的组合。可以理解，本发明所使用的术语“第一”、“第二”、“第三”和“第四”等，仅用于将各组训练样本集进行区分，但各组训练样本集并不受这些术语限制。

本实施例中，以样本图像训练数据作为第一组训练样本集，将样本图像训练数据，有机地与辅助样本图像数据中的预处理图像数据、无缺产品图像数据和光照变换图像数据进行组合，得到第二组训练样本集、第三组训练样本集和第四组训练样本集，提高了得到训练样本集的效率。

如图5所示，在一个实施例中，步骤208具体还包括得到缺陷识别模型的步骤，该步骤具体包括如下步骤：

步骤502，分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型。

其中，缺陷预判模型是初始模型经过训练得到的模型。

具体地，终端分别将每组训练样本集输入对应的初始模型，初始模型根据输入的训练样本集进行训练。终端可以将训练样本集分为多份，训练时，先将一份训练样本集输入初始模型，得到初始模型输出的缺陷预测结果，将缺陷预测结果与缺陷标注数据进行对比并计算预测误差，根据预测误差调整模型参数；再将另一份训练样本集输入调整后的模型，重复上述过程，直至预测误差收敛，得到缺陷预判模型。

步骤504，根据多个缺陷预判模型构建缺陷识别模型。

具体地，终端通过训练得到多个缺陷预判模型后，以多个缺陷预判模型作为子模型，组建缺陷预判模型集群，将得到的缺陷预判模型集群作为母模型，得到缺陷识别模型。

本实施例中，分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型，再根据多个缺陷预判模型构建缺陷识别模型，提高了构建缺陷识别模型的效率。

如图6所示，在一个实施例中，步骤502具体还包括得到缺陷预判模型的步骤，该步骤具体包括如下步骤：

步骤602，对于每组训练样本集，提取训练样本集中的模型输入数据和缺陷标注数据。

其中，缺陷标注数据是样本图像训练数据的样本图像中缺陷的标注数据，是模型训练时模型的期望输出。缺陷可以包括气泡、划痕、杂质、缺损、多余中的至少一种。

具体地，对于第一组训练样本集，终端提取样本图像训练数据中的样本图像及缺陷标注数据，将样本图像作为模型输入数据；对于第二组训练样本集，终端提取样本图像训练数据中的样本图像及缺陷标注数据、预处理图像数据中的预处理图像及标注数据，将样本图像和预处理图像作为模型输入数据；对于第三组训练样本集，终端提取样本图像训练数据中的样本图像及缺陷标注数据、无缺产品图像数据中的无缺产品图像和标注数据，将样本图像和无缺产品图像作为模型输入数据；对于第四组训练样本集，终端提取样本图像训练数据中的样本图像及缺陷标注数据、光照变换图像数据中的光照变换图像和标注数据，将样本图像和光照变换图像作为模型输入数据。

举例说明，终端从样本图像训练数据中提取到的样本图像如图7所示。终端从预处理图像数据中提取到的预处理图像如图8所示，其中预处理图像是经过视觉轮廓提取得到的。终端从无缺产品图像数据中提取到的无缺产品图像如图9所示。缺陷标注数据如图10所示，其中缺陷1002和缺陷1004为标注出位置的缺陷。

步骤604，将模型输入数据输入对应的初始模型，得到初始模型输出的缺陷预测结果。

其中，缺陷预测结果是处于训练过程中的初始模型对模型输入数据中缺陷的预测结果。

具体地，终端预设有与各训练样本集对应的初始模型。终端从训练样本集中提取到模型输入数据与缺陷标注数据后，将模型输入数据输入对应的初始模型，初始模型对模型输入数据进行处理得到缺陷预测结果。

步骤606，根据缺陷预测结果与缺陷标注数据，确定预测误差。

其中，预测误差是以缺陷预测结果与缺陷标注数据之间的差异为自变量的函数。

具体地，终端采用监督学习的方法进行模型训练。终端得到缺陷预测结果后，比较缺陷预测结果与缺陷标注数据，按照预设的误差公式进行计算，得到预测误差。

步骤608，根据预测误差对初始模型进行调整，直至预测误差满足训练停止条件，得到训练样本集对应的缺陷预判模型。

其中，训练停止条件是停止模型训练的条件，训练停止条件可以是预测误差小于预定的误差阈值。

具体地，终端获取预定的误差阈值，比较预测误差与误差阈值。当预测误差大于等于误差阈值时，终端按照减小预测误差的方向，调整初始模型中的模型参数。终端每次对初始模型进行参数调整后，重新对模型输入数据进行处理得到缺陷预测结果，根据缺陷预测结果和缺陷标注数据得到预测误差，比较预测误差与误差阈值，若预测误差仍大于等于误差阈值，再次对模型进行调整，如此循环迭代，直至预测误差小于误差阈值时，停止训练，将停止训练时的模型作为与训练样本集对应的缺陷预判模型。

在一个实施例中，训练停止条件可以是模型训练中预设的迭代次数。当模型在训练中的迭代次数大于等于迭代次数阈值时，终端停止训练，得到缺陷预判模型。

本实施例中，对于每组训练样本集，提取训练样本集中的模型输入数据和缺陷标注数据，将模型输入数据输入对应的初始模型，得到初始模型输出的缺陷预测结果，根据缺陷预测结果与缺陷标注数据，确定预测误差，根据预测误差对初始模型进行调整，当预测误差满足训练停止条件时停止训练，得到训练样本集对应的缺陷预判模型，通过训练不断对初始模型进行调整，提高了获取缺陷预判模型的准确性。

如图11所示，在一个实施例中，步骤208之后还包括使用缺陷识别模型的步骤，该步骤具体包括如下步骤：

步骤1102，获取待检测产品的产品图像数据。

其中，产品图像数据是缺陷识别模型在使用时，输入缺陷识别模型中的待检测产品的图像数据。

具体地，终端获取触发的产品检测指令，根据产品检测指令从存储空间中提取待检测产品的产品图像数据。

在一个实施例中，终端安装有图像采集装置。终端获取产品检测指令后，启动图像采集装置，对流水线上的待检测产品进行图像采集，得到产品图像数据。产品图像数据可以包括多个待检测产品各自对应的图像数据，每个待检测产品可以包括多份图像数据。

步骤1104，将产品图像数据输入缺陷识别模型。

具体地，终端按照缺陷识别模型中缺陷预判模型的数目，对获取到的产品图像数据进行复制，使得每一个缺陷预判模型都对应有相同的产品图像数据。终端对产品图像数据复制完成后，依次将产品图像数据输入缺陷识别模型中的各缺陷预判模型。

步骤1106，获取缺陷识别模型中各缺陷预判模型分别输出的缺陷预判结果。

其中，缺陷预判结果是缺陷预判模型对产品图像数据中的缺陷的识别结果。

具体地，终端将产品图像数据输入缺陷识别模型中的各缺陷预判模型后，各缺陷预判模型对产品图像数据进行处理，识别产品图像数据中的缺陷，分别输出各自的缺陷预判结果。终端获取各缺陷预判模型的缺陷预判结果，以便进行分析处理。

步骤1108，根据获取到的缺陷预判结果确定缺陷识别结果。

其中，缺陷识别结果是缺陷识别模型对产品图像数据中的缺陷的识别结果。

具体地，终端提取各缺陷预判模型的缺陷预判结果，并统计各缺陷预判结果的出现频率，筛选出现频率最高的缺陷预判结果，将筛选得到的出现频率最高的缺陷预判结果，作为缺陷识别模型的缺陷识别结果。缺陷识别结果如图12所示，可以是缺陷部分的图像提取，其中缺陷1202和缺陷1204为识别到的缺陷。在一个实施例中，若终端无法筛选到出现频率最高的缺陷预判结果，比如仅有两种缺陷预判结果且出现概率相同，生成缺陷识别失败的信息，将缺陷识别失败的信息通过显示屏进行显示。

在一个实施例中，终端将筛选到的缺陷预判结果的出现频率与预设的频率阈值进行比较，若缺陷预判结果的出现频率大于等于预设的频率阈值，终端确认筛选得到的缺陷预判结果有效，将筛选得到的缺陷预判结果作为缺陷识别模型的缺陷识别结果。

在一个实施例中，当终端只获取到一种辅助样本图像数据时，终端对一个初始模型训练完成后得到缺陷识别模型。终端在使用缺陷识别模型时，将待检测产品的产品图像数据输入缺陷识别模型后，直接得到缺陷识别模型输出的缺陷识别结果。

本实施例中，获取待检测产品的产品图像数据，将产品图像数据输入缺陷识别模型，获取缺陷识别模型中各缺陷预判模型分别输出的缺陷预判结果，根据各缺陷预判结果确定缺陷识别模型的缺陷识别结果，提高了获取缺陷识别结果的准确性。

图13为一个实施例中缺陷识别模型训练方法的示意图。具体地，参照图10，第一组训练样本集可以由样本图像训练数据组成，第二组训练样本集可以由样本图像训练数据和预处理图像数据组成，第三组训练样本集可以由样本图像训练数据和无缺产品图像数据组成，第四组训练样本集可以由样本图像训练数据和光照变换图像数据组成。终端提取训练样本集中的模型输入数据和缺陷标注数据，以模型输入数据作为模型的输入、缺陷标注数据作为模型的输出，训练各初始模型。

应该理解的是，虽然图2-5和11的流程图中的各个步骤按照箭头的指示依次显示，但是这些步骤并不是必然按照箭头指示的顺序依次执行。除非本文中有明确的说明，这些步骤的执行并没有严格的顺序限制，这些步骤可以以其它的顺序执行。而且，图2-5和11中的至少一部分步骤可以包括多个子步骤或者多个阶段，这些子步骤或者阶段并不必然是在同一时刻执行完成，而是可以在不同的时刻执行，这些子步骤或者阶段的执行顺序也不必然是依次进行，而是可以与其它步骤或者其它步骤的子步骤或者阶段的至少一部分轮流或者交替地执行。

在一个实施例中，如图14所示，提供了一种缺陷识别模型训练装置1400，包括：数据获取模块1402、辅助获取模块1404、样本集构建模块1406和模型得到模块1408，其中：

数据获取模块1402，用于获取样本图像训练数据。

辅助获取模块1404，用于获取与样本图像训练数据对应的辅助样本图像数据。

样本集构建模块1406，用于以样本图像训练数据和辅助样本图像数据，构建训练样本集。

模型得到模块1408，用于根据训练样本集对初始模型进行训练，得到缺陷识别模型。

本实施例中，通过获取样本图像训练数据以及与样本图像训练数据对应的辅助样本图像数据，无需额外对样本图像训练数据提取特征数据，以样本图像训练数据和辅助样本图像数据,构建训练样本集，根据训练样本集对初始模型进行训练,得到缺陷识别模型。在训练时，通过输入与样本图像训练数据对应的辅助样本图像数据，减少了模型训练时提取特征数据所耗费的时间，提高了模型的训练效率。

在一个实施例中，辅助获取模块1404具体包括：类型确定模块、地址获取模块、数据提取模块和辅助确定模块，其中：

类型确定模块，用于确定样本图像训练数据对应的辅助数据类型。

地址获取模块，用于获取各辅助数据类型分别对应的样本数据地址。

数据提取模块，用于根据样本数据地址分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据。

辅助确定模块，用于以预处理图像数据、无缺产品图像数据和光照变换图像数据，作为各种辅助样本图像数据。

本实施例中，通过确定样本图像训练数据对应的辅助数据类型，获取各辅助数据类型分别对应的样本数据地址，根据样本数据地址分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据，再分别将提取到的预处理图像数据、无缺产品图像数据和光照变换图像数据作为辅助样本图像数据，提高了获取辅助样本图像数据的准确性。

在一个实施例中，样本集构建模块1406还用于以样本图像训练数据作为第一组训练样本集；将样本图像训练数据，分别与各种辅助样本图像数据中的预处理图像数据、无缺产品图像数据和光照变换图像数据进行组合，得到第二组训练样本集、第三组训练样本集和第四组训练样本集。

本实施例中，以样本图像训练数据作为第一组训练样本集，将样本图像训练数据，有机地与辅助样本图像数据中的预处理图像数据、无缺产品图像数据和光照变换图像数据进行组合，得到第二组训练样本集、第三组训练样本集和第四组训练样本集，提高了得到训练样本集的效率。

在一个实施例中，模型得到模块1408还用于分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型；根据多个缺陷预判模型构建缺陷识别模型。

本实施例中，分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型，再根据多个缺陷预判模型构建缺陷识别模型，提高了构建缺陷识别模型的效率。在一个实施例中，模型得到模块还用于对于每组训练样本集，提取训练样本集中的模型输入数据和缺陷标注数据；将模型输入数据输入对应的初始模型，得到初始模型输出的缺陷预测结果；根据缺陷预测结果与缺陷标注数据，确定预测误差；根据预测误差对初始模型进行调整，直至预测误差满足训练停止条件，得到训练样本集对应的缺陷预判模型。

本实施例中，对于每组训练样本集，提取训练样本集中的模型输入数据和缺陷标注数据，将模型输入数据输入对应的初始模型，得到初始模型输出的缺陷预测结果，根据缺陷预测结果与缺陷标注数据，确定预测误差，根据预测误差对初始模型进行调整，当预测误差满足训练停止条件时停止训练，得到训练样本集对应的缺陷预判模型，通过训练不断对初始模型进行调整，提高了获取缺陷预判模型的准确性。

在一个实施例中，缺陷识别模型训练装置1400还包括模型使用模块，模型使用模块用于获取待检测产品的产品图像数据；将产品图像数据输入缺陷识别模型；获取缺陷识别模型中各缺陷预判模型分别输出的缺陷预判结果；根据获取到的缺陷预判结果确定缺陷识别结果。

本实施例中，获取待检测产品的产品图像数据，将产品图像数据输入缺陷识别模型，获取缺陷识别模型中各缺陷预判模型分别输出的缺陷预判结果，根据各缺陷预判结果确定缺陷识别模型的缺陷识别结果，提高了获取缺陷识别结果的准确性。

关于缺陷识别模型训练装置的具体限定可以参见上文中对于缺陷识别模型训练方法的限定，在此不再赘述。上述缺陷识别模型训练装置中的各个模块可全部或部分通过软件、硬件及其组合来实现。上述各模块可以硬件形式内嵌于或独立于计算机设备中的处理器中，也可以以软件形式存储于计算机设备中的存储器中，以便于处理器调用执行以上各个模块对应的操作。

在一个实施例中，提供了一种计算机设备，该计算机设备可以是终端，其内部结构图可以如图15所示。该计算机设备包括通过系统总线连接的处理器、存储器、网络接口、显示屏、输入装置和图像采集装置。其中，该计算机设备的处理器用于提供计算和控制能力。该计算机设备的存储器包括非易失性存储介质、内存储器。该非易失性存储介质存储有操作系统和计算机程序。该内存储器为非易失性存储介质中的操作系统和计算机程序的运行提供环境。该计算机设备的网络接口用于与外部的终端通过网络连接通信。该计算机程序被处理器执行时以实现一种缺陷识别模型训练方法。该计算机设备的显示屏可以是液晶显示屏或者电子墨水显示屏，该计算机设备的输入装置可以是显示屏上覆盖的触摸层，也可以是计算机设备外壳上设置的按键、轨迹球或触控板，还可以是外接的键盘、触控板或鼠标等。图像采集装置用于采集待检测产品的产品图像数据。

本领域技术人员可以理解，图15中示出的结构，仅仅是与本申请方案相关的部分结构的框图，并不构成对本申请方案所应用于其上的计算机设备的限定，具体的计算机设备可以包括比图中所示更多或更少的部件，或者组合某些部件，或者具有不同的部件布置。

在一个实施例中，提供了一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，处理器执行计算机程序时实现以下步骤：获取样本图像训练数据；获取与样本图像训练数据对应的辅助样本图像数据；以样本图像训练数据和辅助样本图像数据，构建训练样本集；根据训练样本集对初始模型进行训练，得到缺陷识别模型。在一个实施例中，获取与样本图像训练数据对应的辅助样本图像数据包括：确定样本图像训练数据对应的辅助数据类型；获取各辅助数据类型分别对应的样本数据地址；根据样本数据地址分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据；以预处理图像数据、无缺产品图像数据和光照变换图像数据，作为各种辅助样本图像数据。

在一个实施例中，以样本图像训练数据和辅助样本图像数据，构建训练样本集包括：以样本图像训练数据作为第一组训练样本集；将样本图像训练数据，分别与各种辅助样本图像数据中的预处理图像数据、无缺产品图像数据和光照变换图像数据进行组合，得到第二组训练样本集、第三组训练样本集和第四组训练样本集。

在一个实施例中，根据训练样本集对初始模型进行训练，得到缺陷识别模型包括：分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型；根据多个缺陷预判模型构建缺陷识别模型。

在一个实施例中，分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型包括：对于每组训练样本集，提取训练样本集中的模型输入数据和缺陷标注数据；将模型输入数据输入对应的初始模型，得到初始模型输出的缺陷预测结果；根据缺陷预测结果与缺陷标注数据，确定预测误差；根据预测误差对初始模型进行调整，直至预测误差满足训练停止条件，得到训练样本集对应的缺陷预判模型。

在一个实施例中，根据多个缺陷预判模型构建缺陷识别模型之后，处理器执行计算机程序时还实现以下步骤：获取待检测产品的产品图像数据；将产品图像数据输入缺陷识别模型；获取缺陷识别模型中各缺陷预判模型分别输出的缺陷预判结果；根据获取到的缺陷预判结果确定缺陷识别结果。

本实施例中，通过获取样本图像训练数据以及与样本图像训练数据对应的辅助样本图像数据，无需额外对样本图像训练数据提取特征数据，以样本图像训练数据和辅助样本图像数据,构建训练样本集，根据训练样本集对初始模型进行训练,得到缺陷识别模型。在训练时，通过输入与样本图像训练数据对应的辅助样本图像数据，减少了模型训练时提取特征数据所耗费的时间，提高了模型的训练效率。

在一个实施例中，提供了一种计算机可读存储介质，其上存储有计算机程序，计算机程序被处理器执行时实现以下步骤：获取样本图像训练数据；获取与样本图像训练数据对应的辅助样本图像数据；以样本图像训练数据和辅助样本图像数据，构建训练样本集；根据训练样本集对初始模型进行训练，得到缺陷识别模型。

在一个实施例中，获取与样本图像训练数据对应的辅助样本图像数据包括：确定样本图像训练数据对应的辅助数据类型；获取各辅助数据类型分别对应的样本数据地址；根据样本数据地址分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据；以预处理图像数据、无缺产品图像数据和光照变换图像数据，作为各种辅助样本图像数据。

在一个实施例中，以样本图像训练数据和辅助样本图像数据，构建训练样本集包括：以样本图像训练数据作为第一组训练样本集；将样本图像训练数据，分别与各种辅助样本图像数据中的预处理图像数据、无缺产品图像数据和光照变换图像数据进行组合，得到第二组训练样本集、第三组训练样本集和第四组训练样本集。

在一个实施例中，根据训练样本集对初始模型进行训练，得到缺陷识别模型包括：分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型；根据多个缺陷预判模型构建缺陷识别模型。

在一个实施例中，分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型包括：对于每组训练样本集，提取训练样本集中的模型输入数据和缺陷标注数据；将模型输入数据输入对应的初始模型，得到初始模型输出的缺陷预测结果；根据缺陷预测结果与缺陷标注数据，确定预测误差；根据预测误差对初始模型进行调整，直至预测误差满足训练停止条件，得到训练样本集对应的缺陷预判模型。

在一个实施例中，根据多个缺陷预判模型构建缺陷识别模型之后，计算机程序被处理器执行时还实现以下步骤：获取待检测产品的产品图像数据；将产品图像数据输入缺陷识别模型；获取缺陷识别模型中各缺陷预判模型分别输出的缺陷预判结果；根据获取到的缺陷预判结果确定缺陷识别结果。

本实施例中，通过获取样本图像训练数据以及与样本图像训练数据对应的辅助样本图像数据，无需额外对样本图像训练数据提取特征数据，以样本图像训练数据和辅助样本图像数据,构建训练样本集，根据训练样本集对初始模型进行训练,得到缺陷识别模型。在训练时，通过输入与样本图像训练数据对应的辅助样本图像数据，减少了模型训练时提取特征数据所耗费的时间，提高了模型的训练效率。

本领域普通技术人员可以理解实现上述实施例方法中的全部或部分流程，是可以通过计算机程序来指令相关的硬件来完成，所述的计算机程序可存储于一非易失性计算机可读取存储介质中，该计算机程序在执行时，可包括如上述各方法的实施例的流程。其中，本申请所提供的各实施例中所使用的对存储器、存储、数据库或其它介质的任何引用，均可包括非易失性和/或易失性存储器。非易失性存储器可包括只读存储器(ROM)、可编程ROM(PROM)、电可编程ROM(EPROM)、电可擦除可编程ROM(EEPROM)或闪存。易失性存储器可包括随机存取存储器(RAM)或者外部高速缓冲存储器。作为说明而非局限，RAM以多种形式可得，诸如静态RAM(SRAM)、动态RAM(DRAM)、同步DRAM(SDRAM)、双数据率SDRAM(DDRSDRAM)、增强型SDRAM(ESDRAM)、同步链路(Synchlink)DRAM(SLDRAM)、存储器总线(Rambus)直接RAM(RDRAM)、直接存储器总线动态RAM(DRDRAM)、以及存储器总线动态RAM(RDRAM)等。

以上实施例的各技术特征可以进行任意的组合，为使描述简洁，未对上述实施例中的各个技术特征所有可能的组合都进行描述，然而，只要这些技术特征的组合不存在矛盾，都应当认为是本说明书记载的范围。

以上所述实施例仅表达了本申请的几种实施方式，其描述较为具体和详细，但并不能因此而理解为对发明专利范围的限制。应当指出的是，对于本领域的普通技术人员来说，在不脱离本申请构思的前提下，还可以做出若干变形和改进，这些都属于本申请的保护范围。因此，本申请专利的保护范围应以所附权利要求为准。

Claims (10)

1.一种缺陷识别模型训练方法，所述方法包括：

获取样本图像训练数据；

获取与所述样本图像训练数据对应的辅助样本图像数据；

以所述样本图像训练数据和所述辅助样本图像数据，构建训练样本集；

根据所述训练样本集对初始模型进行训练，得到缺陷识别模型。

2.根据权利要求1所述的方法，其特征在于，所述获取与所述样本图像训练数据对应的辅助样本图像数据包括：

确定所述样本图像训练数据对应的辅助数据类型；

获取各辅助数据类型分别对应的样本数据地址；

根据所述样本数据地址分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据；

以所述预处理图像数据、所述无缺产品图像数据和所述光照变换图像数据，作为各种辅助样本图像数据。

3.根据权利要求2所述的方法，其特征在于，所述以所述样本图像训练数据和所述辅助样本图像数据，构建训练样本集包括：

以所述样本图像训练数据作为第一组训练样本集；

将所述样本图像训练数据，分别与所述各种辅助样本图像数据中的预处理图像数据、无缺产品图像数据和光照变换图像数据进行组合，得到第二组训练样本集、第三组训练样本集和第四组训练样本集。

4.根据权利要求3所述的方法，其特征在于，所述根据所述训练样本集对初始模型进行训练，得到缺陷识别模型包括：

分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型；

根据所述多个缺陷预判模型构建缺陷识别模型。

5.根据权利要求4所述的方法，其特征在于，所述分别根据每组训练样本集对各初始模型进行训练,得到多个缺陷预判模型包括：

对于每组训练样本集，提取所述训练样本集中的模型输入数据和缺陷标注数据；

将所述模型输入数据输入对应的初始模型，得到所述初始模型输出的缺陷预测结果；

根据所述缺陷预测结果与所述缺陷标注数据，确定预测误差；

根据所述预测误差对所述初始模型进行调整，直至所述预测误差满足训练停止条件，得到所述训练样本集对应的缺陷预判模型。

6.根据权利要求1-5中任一项所述的方法，其特征在于，所述根据所述多个缺陷预判模型构建缺陷识别模型之后，还包括：

获取待检测产品的产品图像数据；

将所述产品图像数据输入所述缺陷识别模型；

获取所述缺陷识别模型中各缺陷预判模型分别输出的缺陷预判结果；

根据获取到的缺陷预判结果确定缺陷识别结果。

7.一种缺陷识别模型训练装置，所述装置包括：

数据获取模块，用于获取样本图像训练数据；

辅助获取模块，用于获取与所述样本图像训练数据对应的辅助样本图像数据；

样本集构建模块，用于以所述样本图像训练数据和所述辅助样本图像数据，构建训练样本集；

模型得到模块，用于根据所述训练样本集对初始模型进行训练，得到缺陷识别模型。

8.根据权利要求7所述的装置，其特征在于，所述辅助获取模块包括：

类型确定模块，用于确定所述样本图像训练数据对应的辅助数据类型；

地址获取模块，用于获取各辅助数据类型分别对应的样本数据地址；

数据提取模块，用于根据所述样本数据地址分别提取预处理图像数据、无缺产品图像数据和光照变换图像数据；

辅助确定模块，用于以所述预处理图像数据、所述无缺产品图像数据和所述光照变换图像数据，作为各种辅助样本图像数据。

9.一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现权利要求1至6中任一项所述方法的步骤。

10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现权利要求1至6中任一项所述的方法的步骤。