图像分类方法、装置、电子设备及存储介质制造方法及图纸

技术编号:23213277 阅读:35 留言:0更新日期:2020-01-31 22:03
本公开关于一种图像分类方法、装置、电子设备及存储介质,涉及计算机技术领域,所述方法包括:将目标图像输入预设数目个图像分类模型,获取预设数目个图像分类模型输出的预设数目个标签,确定预设数目个标签的信息熵,若信息熵大于或者等于预设阈值,则确定预设数目个标签的上级标签为目标图像的分类标签。采用本公开,电子设备可以准确确定目标图像的分类标签,该分类标签可以准确体现目标图像的分类结果。

Image classification method, device, electronic equipment and storage medium

【技术实现步骤摘要】
图像分类方法、装置、电子设备及存储介质
本公开涉及计算机领域,尤其涉及一种图像分类方法、装置、电子设备及存储介质。
技术介绍
目前,图像识别技术越来越多的应用于各个领域,其中包括根据图像识别技术识别图像中的物体的种类。相关技术中,电子设备可以根据图像分类模型,确定图像的分类标签。例如,图像A中存在一只哈士奇,则电子设备可以根据图像分类模型,确定图像A的分类标签为“哈士奇”。然而,对于不好分辨的物体,例如上述图像A中的哈士奇,存在一种与哈士奇十分相似的狗类:爱斯基摩犬,则电子设备可能会确定错误的分类标签,即电子设备有可能确定图像A的分类标签为“爱斯基摩犬”,因此会导致分类结果不准确。
技术实现思路
本公开提供一种图像分类方法、装置、电子设备及存储介质,以至少解决相关技术中对图像分类的结果不准确的问题。本公开的技术方案如下:根据本公开实施例的第一方面,提供一种图像分类方法,包括:将目标图像输入预设数目个图像分类模型,获取所述预设数目个图像分类模型输出的预设数目个标签;<br>确定所述预设数本文档来自技高网...

【技术保护点】
1.一种图像分类方法,其特征在于,所述方法包括:/n将目标图像输入预设数目个图像分类模型,获取所述预设数目个图像分类模型输出的预设数目个标签;/n确定所述预设数目个标签的信息熵,所述信息熵表示所述预设数目个标签与所述目标图像的相符程度;/n若所述信息熵大于或者等于预设阈值,则根据预设树状标签结构,确定所述预设数目个标签的公共上级标签为所述目标图像的分类标签,所述预设树状标签结构中级别越高的标签所表示的分类粒度越大,且同一级别的标签所表示的分类粒度相同。/n

【技术特征摘要】
1.一种图像分类方法,其特征在于,所述方法包括:
将目标图像输入预设数目个图像分类模型,获取所述预设数目个图像分类模型输出的预设数目个标签;
确定所述预设数目个标签的信息熵,所述信息熵表示所述预设数目个标签与所述目标图像的相符程度;
若所述信息熵大于或者等于预设阈值,则根据预设树状标签结构,确定所述预设数目个标签的公共上级标签为所述目标图像的分类标签,所述预设树状标签结构中级别越高的标签所表示的分类粒度越大,且同一级别的标签所表示的分类粒度相同。


2.根据权利要求1所述图像分类方法,其特征在于,所述确定所述预设数目个标签的信息熵步骤,包括:
根据所述预设数目个标签建立概率向量,所述概率向量包括所述预设数目个标签中,各相同的标签在所述预设数目个标签中出现的概率;
确定所述概率向量对应的信息熵。


3.根据权利要求2所述图像分类方法,其特征在于,在所述确定所述预设数目个标签的信息熵步骤之后,所述方法还包括:
若所述信息熵小于预设阈值,则确定所述预设数目个标签中出现概率最大的标签为所述目标图像的分类标签。


4.根据权利要求1所述图像分类方法,其特征在于,所述根据预设树状标签结构,确定所述预设数目个标签的公共上级标签为所述目标图像的分类标签步骤,包括:
根据预设树状标签结构,确定所述预设数目个标签在所述预设树状标签结构中最低一级的公共上级标签;
确定所述最低一级的公共上级标签为所述目标图像的分类标签。


5.一种图像分类装置,其特征在于,所述装置包括:
获取单元,被配置为执行将目标图像输入预设数目个图像分类模型,获取所述预设数目个图像分类模型输出的预设数目个...

【专利技术属性】
技术研发人员:张志伟李焱吴丽军
申请(专利权)人:北京达佳互联信息技术有限公司
类型:发明
国别省市:北京;11

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1