[发明专利]一种应用于目标物的自动分类的模型训练方法、装置及电子设备

权利要求书

1.一种应用于目标物的自动分类的模型训练方法，所述方法包括：

将图像样本作为输入数据，将候选目标物作为监督数据训练第一模型，所述第一模型用于从所述图像样本中检测出候选目标物；

基于候选目标物，根据一预设规则确定待分类目标物；

将所述待分类目标物作为输入数据，将所述待分类目标物的类别信息作为监督数据训练第二模型，所述第二模型用于对所述待分类目标物进行分类；

根据所述第一模型和所述第二模型确定用于目标物的自动分类的模型。

2.如权利要求1所述的方法，其中，基于候选目标物，根据一预设规则确定待分类目标物包括：

确定所述候选目标物的图像质量是否满足预设规则；

将满足所述预设规则的候选目标物确定为待分类目标物。

3.如权利要求2所述的方法，其中所述确定所述候选目标物的图像质量是否满足预设规则包括：

确定所述候选目标物的图像的像素和变形程度中的至少一个是否满足预定规则。

4.如权利要求3所述的方法，其中，所述方法还包括：

从输入图像中裁剪出所述候选目标物；

将裁剪出的所述候选目标物转换为预设尺寸；

所述基于候选目标物，根据一预设规则确定待分类目标物，包括：

基于具有所述预设尺寸的候选目标物，根据一预设规则确定待分类目标物。

5.如权利要求1-4任一项所述的方法，其中，所述方法还包括：根据分类结果，调整所述预设规则。

6.一种目标物的自动分类方法，所述方法包括：

利用所述权利要求1-4训练出的用于目标物的自动分类的模型，对输入图像中的目标物进行分类，确定所述目标物的类别。

7.一种应用于目标物的自动分类的模型训练装置，包括：

第一模型训练模块，用于将图像样本作为输入数据，将候选目标物作为监督数据训练第一模型，所述第一模型用于从所述图像样本中检测出候选目标物；

待分类目标物确定模块，用于基于候选目标物，根据一预设规则确定待分类目标物；

第二模型训练模块，用于将所述待分类目标物作为输入数据，将所述待分类目标物的类别信息作为监督数据训练第二模型，所述第二模型用于对所述待分类目标物进行分类；

用于目标物的自动分类的模型确定模块，用于根据所述第一模型和所述第二模型确定用于目标物的自动分类的模型。

8.如权利要求7所述的装置，其中，所述待分类目标物确定模块包括：

图像质量确定单元，用于确定所述候选目标物的图像质量是否满足预设规则；

待分类目标物确定单元，用于将满足所述预设规则的候选目标物确定为待分类目标物。

9.一种计算机可读存储介质，所述存储介质存储有计算机程序，所述计算机程序用于执行上述权利要求1-6任一所述的方法。

10.一种电子设备，所述电子设备包括：

处理器；

用于存储所述处理器可执行指令的存储器；

所述处理器，用于执行上述权利要求1-6任一所述的方法。