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

说明书

本申请实施例提供了一种图像分组方法、装置、电子设备及存储介质，涉及图像处理技术领域，包括：获得待分组图像；计算所述待分组图像与各个图像组中每一图像之间的相似度，其中，每个图像组中包括一对象的至少一图像；若存在第一类图像、且所述第一类图像不属于同一图像组，则将所述第一类图像所属的图像组及所述待分组图像合并为同一图像组，其中，所述第一类图像为：各个图像组中与所述待分组图像的相似度大于等于预设的第一阈值的图像。应用本申请实施例提供的方案可以提高图像分组的准确度。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种图像分组方法、装置、电子设备及存储介质。

背景技术

在图像处理中，为便于对图像进行管理，通常需要对图像进行分组，将相似的图像划分为同一图像组。例如，在上述图像为人脸图像的情况下，认为人脸相似的图像为同一用户的人脸图像，因此可以将人脸相似的图像划分为同一图像组，这样可以使得每一图像组所包含的图像为同一用户的人脸图像。

现有技术中，在进行图像分组时，首先针对每一图像组，根据该图像组所包含的图像确定该图像组对应的聚类中心，得到该图像组的中心图像。然后计算待分组图像与各个图像组的中心图像之间的相似度，确定与待分组图像相似度最高的中心图像对应的图像组，将该待分组图像分配至上述所确定的图像组中。

应用现有技术虽然可以实现对图像进行分组，但是在得到图像组的中心图像时，由于所得到的中心图像难以包含图像组中所有图像的特征，从而可能导致依据中心图像进行图像分组的准确度较低。

发明内容

本申请实施例的目的在于提供一种图像分组方法、装置、电子设备及存储介质，以提高图像分组的准确度。具体技术方案如下：

第一方面，本申请实施例提供了一种图像分组方法，所述方法包括：

获得待分组图像；

计算所述待分组图像与各个图像组中每一图像之间的相似度，其中，每个图像组中包括一对象的至少一图像；

若存在第一类图像、且所述第一类图像不属于同一图像组，则将所述第一类图像所属的图像组及所述待分组图像合并为同一图像组，其中，所述第一类图像为：各个图像组中与所述待分组图像的相似度大于等于预设的第一阈值的图像。

本申请的一个实施例中，所述方法还包括：

若存在第二类图像且不存在所述第一类图像，在所述第二类图像不属于同一图像组的情况下，确定所述第二类图像中与所述待分组图像相似度最高的图像所属的图像组，将所述待分组图像合并至所确定的图像组，其中，所述第二类图像为：各个图像组中与所述待分组图像的相似度大于等于预设的第二阈值且小于所述第一阈值的图像；或

若存在第二类图像且不存在所述第一类图像，在所述第二类图像属于同一图像组的情况下，将所述待分组图像合并至所述第二类图像所属的图像组；或

若存在所述第一类图像、且所述第一类图像属于同一图像组，将所述待分组图像合并至所述第一类图像所属的图像组。

本申请的一个实施例中，所述方法还包括：

获得所述第一类图像所属的图像组中满足标识条件的图像组的组标识，作为目标标识；

将合并后的图像组中各个图像的组标识设置为所述目标标识，其中，每一图像的组标识为：该图像所属图像组的标识。

本申请的一个实施例中，将以下任一图像组作为满足所述标识条件的图像组：

所述第一类图像中与所述待分组图像之间的相似度最高的图像所属的图像组；

包含所述第一类图像最多的图像组；