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

权利要求书

1.一种图像分类方法，其特征在于，所述方法包括：

对待分类图像进行聚类，提取属于同一类别的图像，作为同类图组；

获取同类图组中各个图像的语义标签，所述图像的语义标签为图像中一个或多个物体分别对应的名称；

根据同类图组中各个图像的语义标签以及环境信息，确定所述同类图组对应的类别。

2.根据权利要求1所述的方法，其特征在于，所述对待分类图像进行聚类，提取属于同一类别的图像，包括：

获取待分类图像拍摄时的环境信息；

通过聚类算法，对获取的环境信息进行聚类；

将聚类获得的同一类别的环境信息对应的图像提取为属于同一类别的图像。

3.根据权利要求1或2所述的方法，其特征在于，所述环境信息包括图像拍摄时的地理坐标以及时间。

4.根据权利要求1所述的方法，其特征在于，所述获取同类图组中各个图像的语义标签，包括：

对于同类图组中的每个图像，将所述图像输入标签标注模型，所述标签标注模型用于识别输入的图像中的物体并对识别的物体进行名称标注；

获取所述标签标注模型输出的所述图像的语义标签。

5.根据权利要求1所述的方法，其特征在于，所述根据同类图组中各个图像的语义标签以及环境信息，确定所述同类图组对应的类别，包括：

将同类图组对应的分类信息组输入分类模型，所述分类信息组包括同类图组中的两条分类信息，其中，将同类图组中各个图像的语义标签作为一条分类信息，将同类图组中各个图像的环境信息作为一条分类信息，所述分类模型用于确定输入其中的分类信息组所属类别；

获取所述分类模型输出的类别，作为所述同类图组对应的类别。

6.根据权利要求5所述的方法，其特征在于，所述将同类图组对应的分类信息组输入分类模型之前，还包括：

判断所述同类图组中图像的数量是否达到预设数量；

若否，在所述分类信息组的每条分类信息中添加指定数量的预设数据，作为输入分类模型的分类信息组，所述指定数量为预设数量与同类图组中图像数量之间的差值。

7.根据权利要求1所述的方法，其特征在于，所述待分类图像为电子设备的相册中的图像，所述同类图组对应的类别为所述同类图组所属的事件。

8.一种图像分类装置，其特征在于，所述装置包括：

聚类模块，用于对待分类图像进行聚类，提取属于同一类别的图像，作为同类图组；

标签获取模块，用于获取同类图组中各个图像的语义标签，所述图像的语义标签为图像中一个或多个物体分别对应的名称；

分类模块，用于根据同类图组中各个图像的语义标签以及环境信息，确定所述同类图组对应的类别。

9.一种电子设备，其特征在于，包括：

一个或多个处理器；

存储器；

一个或多个程序，其中所述一个或多个程序被存储在所述存储器中并被配置为由所述一个或多个处理器执行，所述一个或多个程序被所述处理器执行用于执行如权利要求1-7任一项所述的方法。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质中存储有程序代码，所述程序代码可被处理器调用执行如权利要求1-7任一项所述的方法。