【技术实现步骤摘要】
图像聚类方法、装置、电子设备及存储介质
[0001]本申请属于图像处理领域,具体涉及一种图像聚类方法、装置、电子设备及存储介质。
技术介绍
[0002]用户使用手机拍摄的图像中有很多是人物图像。用户为了查找人物图像方便,可能希望查阅同一人物对象的所有图像。
[0003]现有技术一般是根据人物图像中人物对象的人脸特征,确定不同人物图像中是否包含同一人物对象,将包含同一人物对象的人物图像聚合在一起。图1为相关技术提供的图像聚类方法流程示意图,如图1所示,相关技术中人物图像聚类的步骤包括:对人物图像进行人脸检测,如图1中的黑色方框,对检测到的人脸进行对齐;从检测的人脸中提取人脸特征;根据人脸特征进行聚类,确定不同人物图像中的人脸是否属于同一人,将人脸属于同一人的人物图像聚合。因此,人脸特征的提取对聚类有着重要影响。
[0004]由于人物图像中人脸特征提取的内容和精度有限,从而导致基于人脸特征的人物图像聚类不准确。
技术实现思路
[0005]本申请实施例的目的是提供一种图像聚类方法、装置、电子设备及存储介质,以解决上述问题。
[0006]第一方面,本申请实施例提供了一种图像聚类方法,该方法包括:
[0007]对至少两张人物图像分别进行人脸特征分析和人体特征分析,确定各所述人物图像的人脸特征信息、与所述人脸特征信息对应的人脸质量信息和人体特征信息;
[0008]基于所述人脸特征信息、所述人脸质量信息和所述人体特征信息,或者所述人脸特征信息和所述人脸质量信息,确定所述人物图 ...
【技术保护点】
【技术特征摘要】
1.一种图像聚类方法,其特征在于,包括:对至少两张人物图像分别进行人脸特征分析和人体特征分析,确定各所述人物图像的人脸特征信息、与所述人脸特征信息对应的人脸质量信息和人体特征信息;基于所述人脸特征信息、所述人脸质量信息和所述人体特征信息,或者所述人脸特征信息和所述人脸质量信息,确定所述人物图像中每两张人物图像之间的人像相似度;基于所述人像相似度对所述人物图像进行图像聚类,得到至少一个人物图像集合。2.根据权利要求1所述的图像聚类方法,其特征在于,所述基于所述人脸特征信息、所述人脸质量信息和所述人体特征信息,或者所述人脸特征信息和所述人脸质量信息,确定所述人物图像中每两张人物图像之间的人像相似度,包括:在所述两张人物图像满足预设条件的情况下,基于所述两张人物图像的人脸特征信息、所述两张人物图像的人脸质量信息和所述两张人物图像的人体特征信息确定所述人像相似度;在所述两张人物图像不满足所述预设条件的情况下,基于所述两张人物图像的人脸特征信息和所述两张人物图像的人脸质量信息确定所述人像相似度;其中,所述预设条件包括:所述两张人物图像的人体特征信息均不为空,且所述两张人物图像的拍摄时间在同一预设时间段内。3.根据权利要求2所述的图像聚类方法,其特征在于,所述在所述两张人物图像不满足所述预设条件的情况下,基于所述两张人物图像的人脸特征信息和所述两张人物图像的人脸质量信息确定所述人像相似度,包括:在所述两张人物图像不满足所述预设条件的情况下,基于所述两张人物图像的人脸特征信息,确定所述人脸特征信息的相似度以作为待调整相似度;基于所述两张人物图像的人脸质量信息调整所述待调整相似度,得到所述人像相似度。4.根据权利要求2所述的图像聚类方法,其特征在于,所述在所述两张人物图像满足预设条件的情况下,基于所述两张人物图像的人脸特征信息、所述两张人物图像的人脸质量信息和所述两张人物图像的人体特征信息确定所述人像相似度,包括:在所述两张人物图像满足所述预设条件的情况下,基于所述两张人物图像的人脸特征信息和所述两张人物图像的人体特征信息,确定所述人脸特征信息和所述人体特征信息的综合相似度以作为待调整相似度;基于所述两张人物图像的人脸质量信息调整所述待调整相似度,得到所述人像相似度。5.根据权利要求4所述的图像聚类方法,其特征在于,所述基于所述两张人物图像的人脸特征信息和所述两张人物图像的人体特征信息,确定所述人脸特征信息和所述人体特征信息的综合相似度,包括:基于所述两张人物图像的人脸特征信息,确定所述人脸特征信息的相似度;基于所述两张人物图像的人体特征信息,确定所述人体特征信息的相似度;将所述人脸特征信息的相似度和所述人体特征信息的相似度求和取平均值;将所述人脸特征信息的相似度和所述平均值进行比较得到的最大的值确定为所述综合相似度。
6.根据权利要求3或4所述的图像聚类方法,所述人脸质量信息包括人脸角度、人脸清晰度和人脸遮挡程度,所述基于所述两张人物图像的人脸质量信息调整所述待调整相似度,得到所述人像相似度,包括:基于所述两张人物图像中每张人物图像的人脸角度、人脸清晰度和人脸遮挡程度,确定每张所述人物图像的人脸特征综合质量信息;将所述两张人物图像的人脸特征综合质量信息归一化处理,将归一化处理后的所述人脸特征综合质量信息映射到预设最低衰减基数和1之间的区间,得到映射后的所述人脸特征综合质量信息;基于映射后的所述人脸特征综合质量信息调整所述待调整相似度,得到所述人像相似度。7.根据权利要求1
‑
5任一项所述的图像聚类方法,其特征在于,所述每两张人物图像包括第一人物图像和第二人物图像,所述基于所述人像相似度对所述人物图像进行图像聚类,得到至少一个人物图像集合,包括:在所述第一人物图像和所述第二人物图像之间的人像相似度大于第一预设阈值的情况下,从所述人物图像中获取所述第一人物图像的M个近似人物图像,M为正整数,所述近似人物图像是所述人物图像中与所述第一人物图像之间的人像相似度大于第二预设阈值的人物图像;在所述第二人物图像与所述第一人物图像的近似人物图像之间人像相似度的平均值大于第三预设阈值的情况下,控制所述两张人物图像聚类到同一个所述人物图像集合,遍历所述人物图像中的每两张人物图像,得到至少一个人物图像集合。8.根据权利要求1
‑
5任一项所述的图像聚类方法,其特征在于,所述人物图像的人脸特征信息和人脸质量信息是通过将所述人物图像输入人脸超分辨率网络模型中,由所述人脸超分辨率网络模型输出得到的,所述人脸超分辨率网络模型在所述对至少两张人物图像分别进行人脸特征分析和人体特征分析,确定各所述人物图像的人脸特征信息、与所述人脸特征信息对应的人脸质量信息和人体特征信息之前获取,获取方式包括:将每个人物图像样本输入预设人脸超分辨率网络模型中,输出每个所述人物图像样本的人脸特征信息和人脸质量信息;确定每个所述人物图像样本的人脸特征信息的损失值和人脸质量信息的损失值;基于每个所述人物图像样本的人脸质量信息对所述人脸特征信息的损失值进行调整,得到所述人脸特征信息调整后的损失值;基于所述人脸质量信息的损失值和所述人脸特征信息调整后的损失值,联合优化训练...
【专利技术属性】
技术研发人员:程林,
申请(专利权)人:维沃移动通信有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。