基于神经网络的图像检测方法和图像检测装置.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN108875504A(43)申请公布日2018.11.23(21)申请号201711107369.2(22)申请日2017.11.10(71)申请人北京旷视科技有限公司地址100190北京市海淀区科学院南路2号A座313室申请人北京迈格威科技有限公司(72)发明人林孟潇张祥雨(74)专利代理机构北京市柳沈律师事务所11105代理人彭久云夏贝贝(51)Int.Cl.G06K9/00(2006.01)G06K9/46(2006.01)G06N3/04(2006.01)权利要求书5页说明书12页附图3页(54)发明名称基于神经网络的图像检测方法和图像检测装置(57)摘要本公开实施例提供了一种基于神经网络的图像检测方法和图像检测装置。所述方法包括：对所述图像进行特征提取，以获得图像特征；基于所述图像特征，检测所述图像中人体的人头区域；基于所述人头区域的检测结果，确定所述图像中的与所述人头区域对应的人体区域，所述人体区域包括所述人头区域和身体区域。本公开实施例相对于传统的检测装置，提高了检测速度和检测效率。CN108875504ACN108875504A权利要求书1/5页1.一种基于神经网络的图像检测方法，所述方法包括：对所述图像进行特征提取，以获得图像特征；基于所述图像特征，检测所述图像中人体的人头区域；基于所述人头区域的检测结果，确定所述图像中的与所述人头区域对应的人体区域，所述人体区域包括所述人头区域和身体区域。2.根据权利要求1所述的方法，其中，基于所述图像特征，检测所述图像中人体的人头区域的步骤包括：将所述图像输入到第一神经网络中，所述第一神经网络用于提取图像中的人头区域；从所述第一神经网络中输出所述图像中的至少一个人头区域候选框。3.根据权利要求2所述的方法，其中，基于所述图像特征，检测所述图像中人体的人头区域的步骤还包括：从所述第一神经网络中输出所述图像的至少一个人头区域候选框的每个的分数，所述分数表示所述区域是人头区域的可能性；将所述分数与预设人头分数阈值进行比较；将分数大于所述预设人头分数阈值的人头区域确定为人头区域。4.根据权利要求1所述的方法，其中，所述基于所述人头区域的检测结果，确定所述图像中的与所述人头区域对应的人体区域的步骤包括：获取基于机器学习得到的人头和身体的相对位置参数；根据所述相对位置参数确定所述图像中的与所述人头区域对应的至少一个人体区域候选框。5.根据权利要求4所述的方法，其中，所述基于所述人头区域的检测结果，确定所述图像中的与所述人头区域对应的人体区域的步骤还包括：获取预设估计参数，所述预设估计参数表示图像中与所述人头区域对应的至少一个人体区域候选框的个数；基于所述预设估计参数，在所述图像中确定与所述人头区域对应的所述个数的人体区域候选框。6.根据权利要求5所述的方法，其中，所述基于所述人头区域的检测结果，确定所述图像中的与所述人头区域对应的人体区域的步骤还包括：确定每个人体区域候选框的分数值，所述分数值表示所述提取的人体区域候选框是人体区域的可能性；基于所述分数值，从所述个数的人体区域候选框中选择至少一个作为所述人体区域。7.根据权利要求6所述的方法，其中，确定每个人体区域候选框的分数值的步骤包括：将所述个数的人体区域候选框图像输入到训练好的第二神经网络，所述第二神经网络用于确定所述每个人体区域候选框图像的分数值；从所述第二神经网络中输出每个所述人体区域候选框对应的所述分数值。8.根据权利要求4所述的方法，其中，所述基于所述人头区域的检测结果，确定所述图像中的与所述人头区域对应的人体区域的步骤还包括：将所述人体区域候选框进行修正，以得到修正后的人体区域。9.根据权利要求8所述的方法，其中，将所述人体区域候选框进行修正的步骤包括：2CN108875504A权利要求书2/5页将所述人体区域候选框图像输入到所述第三神经网络中，所述第三神经网络用于修正所述人体区域候选框；从所述第三神经网络输出每个所述人体区域候选框的修正结果。10.根据权利要求9所述的方法，其中，从所述第三神经网络输出每个所述人体区域候选框的修正结果的步骤包括：所述第三神经网络基于所述人体区域候选框确定所述人体区域候选框在所述图像中的原始区域；确定所述原始区域是否完整；当所述原始区域不完整时，对该原始区域对应的人体区域候选框进行修正。11.根据权利要求10所述的方法，其中，当所述原始区域不完整时，对该原始区域对应的人体区域候选框进行修正的步骤包括：获取训练好的所述第三神经网络输出的多个标准身体框；将所述人体区域候选框与所述多个标准身体框进行匹配；将匹配率最高的标准身体框作为对应该人体区域候选框的修正框；基于该修正框来对所述对应的人体区域候选框进行修正。12.