[发明专利]人头人体联合检测方法、装置和电子设备

说明书

本发明提供了一种人头人体联合检测方法、装置和电子设备，方法包括：获取多个人头候选框和多个人体候选框；利用至少一个神经网络模型对多个人头候选框和多个人体候选框进行检测，分别得到至少一个人头检测框和至少一个人体检测框；利用至少一个人头检测框中的每个人头检测框和至少一个人体检测框中的任一人体检测框的结构关系，确定人头检测结果和人体检测结果。通过将人头人体分开检测，并对人头、人体的检测结果通过人头检测框和人体检测框的结构关系以及人体检测框得分进行进一步精确的比较分析，能够实现在密集人群场景下，精确检测人头和人体，避免人头被虚检以及人体被漏检。

技术领域

本发明涉及图像处理领域，特别是涉及一种人头人体联合检测方法、装置和电子设备。

背景技术

目前主流的行人检测算法主要基于R-CNN框架，主要包括两部分：第一阶段区域候选网络RPN产生目标候选框；第二阶段区域卷积神经网络R-CNN对候选框进一步地分类和回归。

针对人头、人体检测，第一阶段是把图片输入到RPN，生成人头人体的候选框；第二阶段R-CNN对人头人体候选框进一步地分类和回归，得到人头的检测结果和人体的检测结果。

上述的人头人体的检测方式，在密集人群场景下，会导致人头容易被虚检，人体容易被漏检。

发明内容

有鉴于此，本发明实施例提供了一种人头人体联合检测方法、装置和电子设备，通过将人头人体分开检测，并对人头、人体的检测结果通过人头检测框和人体检测框的结构关系以及人体检测框得分进行进一步精确的比较分析，能够实现在密集人群场景下，精确检测人头和人体，避免人头被虚检以及人体被漏检。

本发明实施例的第一方面在于提供一种人头人体联合检测方法，该方法由设备的处理器执行，该方法包括：获取多个人头候选框和多个人体候选框；利用至少一个神经网络模型对多个人头候选框和多个人体候选框进行检测，分别得到至少一个人头检测框和至少一个人体检测框；利用至少一个人头检测框中的每个人头检测框和至少一个人体检测框中的任一人体检测框的结构关系，确定人头检测结果和人体检测结果。

在一些实施例中，利用至少一个人头检测框中的每个人头检测框和至少一个人体检测框中的任一人体检测框的结构关系，确定第二人头检测结果和第二人体检测结果，包括：当至少一个人头检测框中的第一人头检测框与至少一个人体检测框中的第N人体检测框的重叠面积与第一人头检测框面积的比值大于第一预设阈值时，将第一人头检测框添加到人头检测结果中。

在一些实施例中，人头人体联合检测方法还包括：利用至少一个神经网络模型对多个人体候选框进行检测，分别得到多个人体检测框的得分，多个人体检测框的得分包括第一人体检测框、第二人体检测框以及第N人体检测框的得分，第一人体检测框与第一人头检测框对应；当比值小于或等于第一预设阈值时，比较第一人体检测框的得分与第二预设阈值的大小；当第一人体检测框的得分高于第二预设阈值时，将第一人体检测框添加到人体检测结果中。

在一些实施例中，人头人体联合检测方法还包括：当第一人体检测框的得分等于或低于第二预设阈值时，将第一人头检测框从人头检测结果中去掉。

在一些实施例中，获取多个人头候选框和多个人体候选框，包括：获取多个人头候选框；利用人头和人体的位置对应关系，根据多个人头候选框确定多个人体候选框。

在一些实施例中，获取多个人头候选框，包括：利用区域候选网络RPN获取多个人头候选框。

在一些实施例中，至少一个神经网络模型包括第一R-CNN模型和第二R-CNN模型。