本申请提供一种图像分类模型训练方法、装置、电子设备及计算机可读存储介质。该图像分类模型训练方法包括：获取目标样本图像；通过预设的热力图映射参数，对所述目标样本图像进行热力信息提取，得到所述目标样本图像的样本热力图，其中，所述热力图映射参数用于反映样本热力图与类别标注信息之间的约束关系，所述热力图映射参数基于预设的热力图提取模块学习得到；基于所述样本热力图，确定所述目标样本图像的类别标注信息；基于所述类别标注信息和所述目标样本图像，对待训练的图像分类模型进行训练，直至符合预设的第一停止训练条件时，得到已训练图像分类模型。本申请中可以提高图像分类模型分类准确度。高图像分类模型分类准确度。高图像分类模型分类准确度。
【技术实现步骤摘要】
图像分类模型训练方法、装置、电子设备及存储介质
[0001]本申请涉及计算机视觉
，具体涉及一种图像分类模型训练方法、装置、电子设备及计算机可读存储介质。
技术介绍
[0002]随着计算机视觉技术的不断发展，计算机视觉技术被应用于越来越多的领域。其中，图像分类是计算机视觉任务中的一项重要任务。
[0003]现有的图像分类模型需要基于大量的数据标注作为监督信息进行训练，才能保证图像分类模型的分类准确率。
[0004]但是，对于某些特殊场景，由于类别间差异太小，人工肉眼无法准确区分图像类别，因而会导致标注错误；而采用错误的人工标注对图像分类模型进行训练而导致的已训练图像分类模型分类混乱的问题，最后将会导致训练后的图像分类模型的分类准确度较低。
技术实现思路
[0005]本申请提供一种图像分类模型训练方法、装置、电子设备及计算机可读存储介质，旨在解决由于类别间差异太小容易采用错误的人工标注对图像分类模型进行训练，进而导致的图像分类模型分类准确度低的问题。
[0006]第一方面，本申请提供一种图像分类模型训练方法，所述方法包括：
[0007]获取目标样本图像；
[0008]通过预设的热力图映射参数，对所述目标样本图像进行热力信息提取，得到所述目标样本图像的样本热力图，其中，所述热力图映射参数用于反映样本热力图与类别标注信息之间的约束关系，所述热力图映射参数基于预设的热力图提取模块学习得到；
[0009]基于所述样本热力图，确定所述目标样本图像的类别标注信息；
[0010]基于所述类别标注信息和所述目标样本图像，对待训练的图像分类模型进行训练，直至符合预设的第一停止训练条件时，得到已训练图像分类模型。
[0011]第二方面，本申请提供一种图像分类模型训练装置，所述图像分类模型训练装置包括：
[0012]获取单元，用于获取目标样本图像；
[0013]提取单元，用于通过预设的热力图映射参数，对所述目标样本图像进行热力信息提取，得到所述目标样本图像的样本热力图，其中，所述热力图映射参数用于反映样本热力图与类别标注信息之间的约束关系，所述热力图映射参数基于预设的热力图提取模块学习得到；
[0014]确定单元，用于基于所述样本热力图，确定所述目标样本图像的类别标注信息；
[0015]训练单元，用于基于所述类别标注信息和所述目标样本图像，对待训练的图像分类模型进行训练，直至符合预设的第一停止训练条件时，得到已训练图像分类模型。
[0016]在本申请的一些实施例中，所述确定单元具体用于：
[0017]获取热力位置信息与类别信息之间的预设映射关系，其中，所述预设映射关系用于指示所述样本热力图的热力位置信息与所述目标样本图像的类别标注信息之间的关系；
[0018]获取所述样本热力图的热力位置信息；
[0019]基于所述样本热力图的热力位置信息和所述预设映射关系，确定所述目标样本图像的类别标注信息。
[0020]在本申请的一些实施例中，所述基于所述类别标注信息和所述目标样本图像，对待训练的图像分类模型进行训练，直至符合预设的第一停止训练条件时，得到已训练图像分类模型之前，所述训练单元具体用于：
[0021]获取图像样本集合以及预设标注数据，其中，所述图像样本集合包括多个初始样本图像，所述预设标注数据包括所述初始样本图像的预设标注类别；
[0022]基于预设分类模型对所述图像样本集合中各初始样本图像进行预测，得到所述图像样本集合中各初始样本图像的预测类别；
[0023]从所述图像样本集合中，获取所述预设标注类别与所述预测类别不同的目标初始样本图像；
[0024]以所述目标初始样本图像的预测类别作为监督信息，基于所述目标初始样本图像对所述预设分类模型进行训练，直至符合预设的第二停止训练条件时，得到所述待训练的图像分类模型。
[0025]在本申请的一些实施例中，所述基于所述样本热力图，确定所述目标样本图像的类别标注信息之前，所述训练单元具体用于：
[0026]通过所述热力图映射参数，对所述目标初始样本图像进行热力信息提取，得到所述目标初始样本图像的目标热力图；
[0027]通过待训练的图像分类模型，根据所述目标初始样本图像进行预测，得到所述目标初始样本图像的目标类别；
[0028]获取所述目标热力图的热力位置信息；
[0029]基于所述目标热力图的热力位置信息和所述目标类别，获取热力位置信息与类别信息之间的预设映射关系。
[0030]在本申请的一些实施例中，所述提取单元具体用于：
[0031]通过所述热力图映射参数，对所述目标样本图像进行特征提取，得到所述目标样本图像的目标特征图；
[0032]基于所述目标特征图，获取所述目标样本图像的样本热力图。
[0033]在本申请的一些实施例中，所述图像分类模型训练装置还包括分类单元，所述分类单元具体用于：
[0034]获取目标场景区域的目标场景图像；
[0035]通过所述已训练图像分类模型，对所述目标场景图像进行分类，得到所述目标场景区域的目标场景状态。
[0036]在本申请的一些实施例中，所述目标场景图像包括所述目标场景区域的在先场景图像和在后场景图像，所述分类单元具体用于：
[0037]基于所述在先样本图像和所述在后样本图像，确定所述目标样本图像的样本光流
信息；
[0038]以所述类别标注信息作为监督信息，基于所述在先样本图像和所述样本光流信息对待训练的图像分类模型进行训练，直至符合预设的第一停止训练条件时，得到已训练图像分类模型；
[0039]在本申请的一些实施例中，所述目标样本图像包括样本场景区域的在先样本图像和在后样本图像，所述训练单元具体用于：
[0040]基于所述在先场景图像和所述在后场景图像，确定所述目标场景图像的场景光流信息；
[0041]通过所述已训练图像分类模型，根据所述在先场景图像和所述场景光流信息，对所述目标场景图像进行分类，得到所述目标场景区域的目标场景状态。
[0042]第三方面，本申请还提供一种电子设备，所述电子设备包括处理器和存储器，所述存储器中存储有计算机程序，所述处理器调用所述存储器中的计算机程序时执行本申请提供的任一种图像分类模型训练方法中的步骤。
[0043]第四方面，本申请还提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器进行加载，以执行所述的图像分类模型训练方法中的步骤。
[0044]本申请中，由于热力图映射参数基于预设的热力图提取模块学习得到，因此热力图映射参数还可以反映指示样本热力图与图像特征信息之间的关系；而特征提取层提取的图像特征信息与图像类别是直接相关的，故热力图映射参数也在一定程度上反映了样本热力图与类别标注信息之间的约束关系。由此可知，通过采用样本热力图对目标样本图像的类别标注信息进行指示，根据样本热力图确本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种图像分类模型训练方法，其特征在于，所述方法包括：获取目标样本图像；通过预设的热力图映射参数，对所述目标样本图像进行热力信息提取，得到所述目标样本图像的样本热力图，其中，所述热力图映射参数用于反映样本热力图与类别标注信息之间的约束关系，所述热力图映射参数基于预设的热力图提取模块学习得到；基于所述样本热力图，确定所述目标样本图像的类别标注信息；基于所述类别标注信息和所述目标样本图像，对待训练的图像分类模型进行训练，直至符合预设的第一停止训练条件时，得到已训练图像分类模型。2.根据权利要求1所述的图像分类模型训练方法，其特征在于，所述基于所述样本热力图，确定所述目标样本图像的类别标注信息，包括：获取热力位置信息与类别信息之间的预设映射关系，其中，所述预设映射关系用于指示所述样本热力图的热力位置信息与所述目标样本图像的类别标注信息之间的关系；获取所述样本热力图的热力位置信息；基于所述样本热力图的热力位置信息和所述预设映射关系，确定所述目标样本图像的类别标注信息。3.根据权利要求1所述的图像分类模型训练方法，其特征在于，所述基于所述类别标注信息和所述目标样本图像，对待训练的图像分类模型进行训练，直至符合预设的第一停止训练条件时，得到已训练图像分类模型之前，还包括：获取图像样本集合以及预设标注数据，其中，所述图像样本集合包括多个初始样本图像，所述预设标注数据包括所述初始样本图像的预设标注类别；基于预设分类模型对所述图像样本集合中各初始样本图像进行预测，得到所述图像样本集合中各初始样本图像的预测类别；从所述图像样本集合中，获取所述预设标注类别与所述预测类别不同的目标初始样本图像；以所述目标初始样本图像的预测类别作为监督信息，基于所述目标初始样本图像对所述预设分类模型进行训练，直至符合预设的第二停止训练条件时，得到所述待训练的图像分类模型。4.根据权利要求3所述的图像分类模型训练方法，其特征在于，所述基于所述样本热力图，确定所述目标样本图像的类别标注信息之前，还包括：通过所述热力图映射参数，对所述目标初始样本图像进行热力信息提取，得到所述目标初始样本图像的目标热力图；通过待训练的图像分类模型，根据所述目标初始样本图像进行预测，得到所述目标初始样本图像的目标类别；获取所述目标热力图的热力位置信息；基于所述目标热力图的热力位置信息和所述目标类别，获取热力位置信息与类别信息之间的预设映射关系。5.根据权利要求1所述的图像分类模型训练方法，其特征在于，所述通过预设的热力图映射参数，对所述目标样本图像进行热力信息提取，得到所述目标样本图像的样本热力图...
【专利技术属性】
技术研发人员：杨小平，陈奇华，苏鹏，
申请(专利权)人：顺丰科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。