[发明专利]一种图片聚类方法、装置及存储介质

说明书

本发明公开了一种图片聚类方法，包括：分别对多张待聚类图片进行特征提取，得到多个第一特征向量；对所述多个第一特征向量进行聚类，得到至少一个第一聚类簇；确定所述至少一个第一聚类簇中每一个第一聚类簇的中心特征向量；将每个第一聚类簇的中心特征向量与中心特征向量所属第一聚类簇中的其他第一特征向量分别进行融合，得到多个第二特征向量；对所述多个第二特征向量进行聚类，得到至少一个第二聚类簇；基于所述至少一个第二聚类簇进行图片聚类。通过上述图片聚类方法，优化了类中心特征的计算方式，优化图片特征的表达方式，提高了聚类的准确率。

技术领域

本发明属于图片聚类技术领域，具体涉及一种图片聚类方法、装置及存储介质。

背景技术

人脸聚类是一项将人脸识别等算法应用在安防领域的重要技术。具体的应用如对人脸进行分组，把相同的人脸分为同一档案，后续可以根据相关的人脸信息进行溯源调查。此外，人脸聚类可以在无注册信息的场景下，对采集到的人脸信息进行建档，后续可以用于分析用户画像、人员监控等。

目前人脸聚类技术主要分为两类方法，一类是在实时抓拍过程中，利用人脸质量分数抓拍多张待聚类图片，再用人脸识别归档聚类。另一类是先对人脸进行特征提取，然后对特征利用聚类算法进行聚类。由于业务需求对准确度更关注，因此现在多使用第二类图片聚类方法。

现有的技术方案，首先训练人脸特征提取模型，接着根据人脸特征计算两两图像的相似度，然后使用聚类算法，最后按照一定的规则、阈值进行类的合并和特殊情况处理。这类方法存在准确度低的缺点，且高度依赖于人脸特征提取模型的效果。其它的方法，也是首先训练人脸特征模型提取人脸特征，在进行聚类，不仅仅使用人脸特征信息，还结合了其它的信息，如技术使用两两图像属于同一类的置信度，基于两两图像的时空匹配度信息和特征相似度，确定其属于同一类别的联合概率。除了两两图像的相似度，还计算了图像与类中心特征的相似度。这类方法虽然准确度有了一定提升，但仍不能满足一些特定业务场景需求，特别是对算法效率要求高的场景。

发明内容

鉴于以上所述现有技术的缺点，本发明提供一种图片聚类方法、装置及存储介质，以优化类中心特征的计算方式。

为实现上述目的及其他相关目的，本发明提供一种图片聚类方法，该方法包括：

分别对多张待聚类图片进行特征提取，得到多个第一特征向量；

对所述多个第一特征向量进行聚类，得到至少一个第一聚类簇；

确定所述至少一个第一聚类簇中每一个第一聚类簇的中心特征向量；

将每个第一聚类簇的中心特征向量与中心特征向量所属第一聚类簇中的其他第一特征向量分别进行融合，得到多个第二特征向量；

对所述多个第二特征向量进行聚类，得到至少一个第二聚类簇；

基于所述至少一个第二聚类簇进行图片聚类。

可选地，确定所述至少一个第一聚类簇中每一个第一聚类簇的中心特征向量，包括：

在每个第一聚类簇中，确定第三特征向量，所述第三特征向量为第一聚类簇中拥有最多相似特征向量的特征向量；

计算第三特征向量与第一聚类簇中其他特征向量的第一相似度，将第一相似度大于设定相似度阈值的其中一个特征向量作为第四特征向量；

对所述第三特征向量与所述第四特征向量进行加权平均，得到第一个新的特征向量；

计算第一个新的特征向量与第一聚类簇中其他特征向量的第二相似度，将第二相似度大于设定相似度阈值的其中一个特征向量作为第五特征向量；

对所述第五特征向量与所述第一个新的特征向量进行加权平均，得到第二个新的特征向量；