[发明专利]用于处理图像的方法和装置

说明书

本公开的实施例公开了用于处理图像的方法和装置。该方法的一具体实施方式包括：获取目标人体图像；从类别集合中确定目标人体图像所属的目标类别，其中，预设人体图像集合中的预设人体图像的类别为预设人体图像集合所对应的类别，预设人体图像集合包括预设人体图像子集合，预设人体图像子集合中的预设人体图像对应相同的人物；获取目标类别对应的候选人体图像集合；确定候选人体图像集合中的预设人体图像子集合所对应的人物与目标人体图像所对应的人物的相似度是否大于或等于目标相似度阈值；响应于确定是，将大于或等于的预设人体图像子集合确定为结果人体图像子集合。该实施方式可以提高图像匹配的精度和效率，减小图像处理所消耗的资源。

技术领域

本公开的实施例涉及计算机技术领域，尤其涉及用于处理图像的方法和装置。

背景技术

行人重识别(Person re-identification)也称行人再识别，是利用计算机视觉技术判断拍摄得到的图像中是否存在特定行人对象的技术。

目前，由于不同摄像设备之间的差异，同时行人兼具刚性和柔性的特性，外观易受穿着、尺度、遮挡、姿态和视角等影响，使得行人重识别成为计算机视觉领域中一个既具有研究价值，又极具挑战性的热门课题。

发明内容

本公开的实施例提出了用于处理图像的方法和装置。

第一方面，本公开的实施例提供了一种用于处理图像的方法，该方法包括：获取目标人体图像；从预先确定的类别集合中确定目标人体图像所属的类别作为目标类别，其中，类别集合中的类别与预设人体图像集合相对应，预设人体图像集合中的预设人体图像的类别为预设人体图像集合所对应的类别，预设人体图像集合包括至少一个预设人体图像子集合，预设人体图像子集合中的预设人体图像对应相同的人物；获取目标类别对应的预设人体图像集合作为候选人体图像集合；确定候选人体图像集合中的预设人体图像子集合所对应的人物与目标人体图像所对应的人物的相似度是否大于或等于目标相似度阈值；响应于确定是，将大于或等于目标相似度阈值的预设人体图像子集合确定为目标人体图像所对应的结果人体图像子集合。

在一些实施例中，类别集合包括基于人体图像是否包含人脸对象划分的包含人脸类和不包含人脸类；以及从预先确定的类别集合中确定目标人体图像所属的类别作为目标类别包括：将目标人体图像输入预先训练的人脸检测模型，获得检测结果，其中，检测结果用于指示目标人体图像是否包含人脸对象；基于检测结果，从类别集合中确定目标人体图像所属的类别作为目标类别。

在一些实施例中，类别集合包括基于人体图像中的人体对象的朝向划分的正面类、侧面类和背面类；以及从预先确定的类别集合中确定目标人体图像所属的类别作为目标类别包括：将目标人体图像输入预先训练的人脸检测模型，获得检测结果，其中，检测结果用于指示目标人体图像是否包含人脸对象；将目标人体图像输入预先训练的人体骨骼关键点识别模型，获得识别结果，其中，识别结果用于指示人体骨骼关键点在目标人体图像中的位置；基于所获得的检测结果和识别结果，从类别集合中确定目标人体图像所属的类别作为目标类别。

在一些实施例中，该方法还包括：将目标人体图像添加到结果人体图像子集合中，获得更新后的结果人体图像子集合。

在一些实施例中，目标人体图像是从目标人体视频中提取的人体图像；以及在将大于或等于目标相似度阈值的预设人体图像子集合确定为目标人体图像所对应的结果人体图像子集合之后，该方法还包括：从目标人体视频中选取目标数量个人体图像；对于目标数量个人体图像中的人体图像，执行以下处理步骤：从类别集合中确定该人体图像所属的类别；响应于该人体图像所属的类别与目标类别相同，将该人体图像添加到结果人体图像子集合中，获得更新后的结果人体图像子集合。

在一些实施例中，处理步骤还包括：响应于该人体图像所属的类别与目标类别不相同，从该人体图像所属的类别所对应的预设人体图像集合中确定与结果人体图像子集合对应同一个人物的预设人体图像子集合；将该人体图像添加到所确定的预设人体图像子集合中，获得更新后的预设人体图像子集合。