[发明专利]一种分类模型的迭代方法、装置及电子设备

权利要求书

1.一种分类模型的迭代方法，其特征在于，包括：

获取待测图像和对应每一所述待测图像的分类结果；

将所有所述待测图像按照预先设置的比例分别存放至训练集、验证集和测试集中；

根据所述训练集和所述训练集中的待测图像对应的分类结果对参考分类模型进行训练；

根据所述验证集和所述验证集中的待测图像对应的分类结果，验证所述参考分类模型和训练后的参考分类模型的准确率，并选择准确率较高的作为最优参考分类模型；

根据所述测试集和所述测试集中的待测图像对应的分类结果，测试所述最优参考分类模型和产线使用的当前分类模型的准确率；

在所述最优参考分类模型的准确率高于所述当前分类模型的准确率的情况下，控制所述最优参考分类模型替换所述当前分类模型。

2.根据权利要求1所述的迭代方法，其特征在于，所述获取待测图像和对应每一待测图像的分类结果的步骤包括：

控制相机采集待测图像；

控制所述参考分类模型对所有所述待测图像进行分类测试，得到每一所述待测图像的分类结果。

3.根据权利要求2所述的迭代方法，其特征在于，所述迭代方法还包括：

控制所述参考分类模型对所有所述待测图像进行分类测试，得到每一所述待测图像的分类结果和对应每一分类结果的置信度；

判断所述置信度是否小于预先设置的置信度阈值，如是，则：

获取用户输入的确认分类决定作为所述置信度对应的分类结果。

4.根据权利要求1所述的迭代方法，其特征在于，所述迭代方法还包括：

按照预先设定的周期获取待测图像和对应每一待测图像的分类结果。

5.根据权利要求1所述的迭代方法，其特征在于，所述迭代方法还包括：

如果所述当前分类模型与所述参考分类模型相同，则将训练后的参考分类模型作为最优参考分类模型。

6.一种分类模型的迭代装置，其特征在于，包括：

获取模块，用于获取待测图像和对应每一所述待测图像的分类结果；

存储模块，用于将所有所述待测图像按照预先设置的比例分别存放至训练集、验证集和测试集中；

训练模块，用于根据所述训练集和所述训练集中的待测图像对应的分类结果对参考分类模型进行训练；

验证模块，用于根据所述验证集和所述验证集中的待测图像对应的分类结果，验证所述参考分类模型和训练后的参考分类模型的准确率，并选择准确率较高的作为最优参考分类模型；

测试模块，用于根据所述测试集和所述测试集中的待测图像对应的分类结果，测试所述最优参考分类模型和产线使用的当前分类模型的准确率；

替换模块，用于在所述最优参考分类模型的准确率高于所述当前分类模型的准确率的情况下，控制所述最优参考分类模型替换所述当前分类模型。

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

采集单元，用于控制相机采集待测图像；

测试单元，用于控制所述参考分类模型对所有所述待测图像进行分类测试，得到每一所述待测图像的分类结果。

8.根据权利要求7所述的迭代装置，其特征在于，所述测试单元还用于控制所述参考分类模型对所有所述待测图像进行分类测试，得到每一所述待测图像的分类结果和对应每一分类结果的置信度；

所述获取模块还包括：

判断单元，用于判断所述置信度是否小于预先设置的置信度阈值；

获取单元，用于在所述判断模块的判断结果为是的情况下，获取用户输入的确认分类决定作为所述置信度对应的分类结果。

9.一种电子设备，其特征在于，包括根据权利要求6-8中任一项所述的迭代装置。

10.一种电子设备，其特征在于，包括存储器和处理器，所述存储器用于存储指令，所述指令用于控制所述处理器执行根据权利要求1-5中任一项所述的迭代方法。