[发明专利]对象计数方法及装置、电子设备和存储介质

说明书

本公开涉及一种对象计数方法及装置、电子设备和存储介质，所述方法包括：对待处理图像进行特征提取，得到所述待处理图像的初始特征图；根据所述初始特征图，得到所述待处理图像中的头部关键点及头部区域；根据所述头部关键点及所述头部区域，对所述待处理图像进行对象计数及定位，确定所述待处理图像中对象的数量及位置。本公开实施例可实现提高对象计数和定位的精度。

技术领域

本公开涉及计算机技术领域，尤其涉及一种对象计数方法及装置、电子设备和存储介质。

背景技术

人群计数技术是对于视频数据采集等场景下的视频，通过计算机视觉算法来评估画面中的实时人数、人的分布情况、人群的密度等信息的技术。

发明内容

本公开提出了一种进行计数及定位的对象计数技术方案。

根据本公开的一方面，提供了一种对象技术方法，包括：

对待处理图像进行特征提取，得到所述待处理图像的初始特征图；

根据所述初始特征图，得到所述待处理图像中的头部关键点及头部区域；

根据所述头部关键点及所述头部区域，对所述待处理图像进行对象计数及定位，确定所述待处理图像中对象的数量及位置。

在一种可能的实现方式中，所述根据所述头部关键点及所述头部区域，对所述待处理图像进行对象计数及定位，确定所述待处理图像中对象的数量及位置，包括：

确定各所述头部关键点与各所述头部区域的中心点之间的相似度；

确定与所述头部关键点的相似度满足阈值条件的头部区域，将所述与所述头部关键点的相似度满足阈值条件的头部区域与所述头部关键点进行合并，得到合并信息；

根据所述合并信息、未合并的头部关键点及未合并的头部区域，对所述待处理图像进行计数及定位，确定所述待处理图像中对象的数量及位置。

在一种可能的实现方式中，所述确定与所述头部关键点的相似度满足阈值条件的头部区域，将所述与所述头部关键点的相似度满足阈值条件的头部区域与所述头部关键点进行合并，得到合并信息，包括：

删除所述与所述头部关键点的相似度满足阈值条件的头部区域；

根据所述头部关键点及所述头部区域的尺寸，得到合并后的头部区域，其中，所述合并信息包括所述头部关键点及所述合并后的头部区域。

在一种可能的实现方式中，所述确定与所述头部关键点的相似度满足阈值条件的头部区域，将所述与所述头部关键点的相似度满足阈值条件的头部区域与所述头部关键点进行合并，得到合并信息，包括：

将所述头部关键点按照置信度进行排序；

按照所述头部关键点的置信度由大到小的顺序，将所述头部关键点与所述头部区域进行合并，得到合并信息。

在一种可能的实现方式中，所述根据所述初始特征图，得到所述待处理图像中的头部关键点，包括：

对所述初始特征图进行下采样处理，得到不同尺度的多个下采样特征图；

对多个所述下采样特征图中的目标下采样特征图进行上采样处理，得到上采样特征图，所述上采样特征图与所述初始特征图尺度一致；

根据所述上采样特征图进行头部关键点识别，得到所述待处理图像中的头部关键点。

在一种可能的实现方式中，所述根据所述初始特征图，得到所述待处理图像中的头部区域，还包括：

分别对多个所述下采样特征图进行头部区域识别，得到各个所述下采样特征图对应的头部区域；