[发明专利]一种图片分类的方法、装置、电子设备及存储介质

说明书

本发明实施例公开了一种图片分类的方法、装置、电子设备及存储介质。其中，该方法包括：获取原始待分类图片，将原始待分类图片转换为张量数据；将原始待分类图片的张量数据输入至预先训练的图片分类模型中，得到预设的至少三种图片类别下原始待分类图片对应的分类概率；判断原始待分类图片是否存在等于或大于预设概率阈值的分类概率，若是，则确定原始待分类图片为目标待分类图片；根据目标待分类图片的分类概率，确定目标待分类图片的候选图片类别，判断候选图片类别是否为预设待确认图片类别；若是，则将目标待分类图片的张量数据输入至预先训练的二分类器中，得到目标分类图片的目标图片类别，实现提高图片分类精度。

技术领域

本发明实施例涉及图像处理技术，尤其涉及一种图片分类的方法、装置、电子设备及存储介质。

背景技术

图片分类可以根据图片中所反映的不同特征，对图片进行类别划分，以代替人的视觉判读。

不同类别的图片之间存在相似的特征，现有的图片分类方法采用神经网络模型提取图片特征，但现有技术无法准确识别不同类型的图片之间的区别。例如，图片类别为立面图、平面图和剖面图，在进行图片分类时，立面图、平面图和剖面图很容易出现互相错分的情况，导致图片分类的精度较低。

发明内容

本发明实施例提供一种图片分类的方法、装置、电子设备及存储介质，以提高图片分类的精度。

第一方面，本发明实施例提供了一种图片分类的方法，该方法包括：

获取原始待分类图片，将所述原始待分类图片转换为张量数据；

将所述原始待分类图片的张量数据输入至预先训练的图片分类模型中，得到预设的至少三种图片类别下所述原始待分类图片对应的分类概率；

判断所述原始待分类图片是否存在等于或大于预设的概率阈值的分类概率，若是，则确定所述原始待分类图片为目标待分类图片；

根据所述目标待分类图片的分类概率，确定所述目标待分类图片的候选图片类别，判断所述候选图片类别是否为预设的待确认图片类别；

若是，则将所述目标待分类图片的张量数据输入至预先训练的二分类器中，得到所述目标分类图片的目标图片类别。

第二方面，本发明实施例还提供了一种图片分类的装置，该装置包括：

数据转换模块，用于获取原始待分类图片，将所述原始待分类图片转换为张量数据；

概率获得模块，用于将所述原始待分类图片的张量数据输入至预先训练的图片分类模型中，得到预设的至少三种图片类别下所述原始待分类图片对应的分类概率；

目标图片确定模块，用于判断所述原始待分类图片是否存在等于或大于预设的概率阈值的分类概率，若是，则确定所述原始待分类图片为目标待分类图片；

候选类别确定模块，用于根据所述目标待分类图片的分类概率，确定所述目标待分类图片的候选图片类别，判断所述候选图片类别是否为预设的待确认图片类别；

目标类别二分类模块，用于若是，则将所述目标待分类图片的张量数据输入至预先训练的二分类器中，得到所述目标分类图片的目标图片类别。

第三方面，本发明实施例还提供了一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现如本发明任意实施例所述的图片分类的方法。

第四方面，本发明实施例还提供了一种包含计算机可执行指令的存储介质，所述计算机可执行指令在由计算机处理器执行时用于执行如本发明任意实施例所述的图片分类的方法。