[发明专利]图像识别的方法、装置、计算设备及介质

说明书

本公开提供了一种图像识别的方法。该方法包括：获取包含人像的原始图像；确定原始图像中的骨骼信息；根据骨骼信息，从原始图像中确定目标区域；以及识别目标区域中的人像是否以预设方式佩戴安全装置。本公开还提供了一种图像识别的装置、一种计算设备及一种介质。

技术领域

本公开涉及计算机技术领域，更具体地，涉及一种图像识别的方法、装置、计算设备及介质。

背景技术

例如安全帽等的安全装置能够对人头部在受坠落物等伤害时起到防护作用，是施工作业时必须佩带的防护品。但很多施工人员由于疏忽或侥幸心理不佩戴安全帽，在事故中造成不必要的伤亡。通过安全帽佩带识别技术，能够及时发现没有佩戴安全帽的施工人员，从而提高施工过程中的管理效率。

传统的安全帽佩戴识别方法一般先将待识别图像中人体的整体轮廓中检测出来，然后在整体轮廓中找出安全帽的位置，以判断图像中的人体是否佩戴安全帽。

在实现本公开构思的过程中，发明人发现上述方法在识别图像的过程中，至少存在识别效率较低、准确性不高的问题。

发明内容

有鉴于此，本公开提供了一种图像识别的方法、装置、计算设备及介质。

本公开的一个方面提供了一种图像识别的方法，包括：获取包含人像的原始图像；确定原始图像中的骨骼信息；根据骨骼信息，从原始图像中确定目标区域；以及识别目标区域中的人像是否以预设方式佩戴安全装置。

根据本公开的实施例，上述确定原始图像中的骨骼信息，包括：通过将原始图像输入预设的人体骨骼关键点检测模型，获取骨骼信息，骨骼信息包括脖子位置、左耳位置和右耳位置的坐标。

根据本公开的实施例，上述根据骨骼信息，从原始图像中确定目标区域，包括：根据左耳位置的坐标和右耳位置的坐标，确定中心点；以中心点作为中心，确定矩形区域，矩形区域的长为中心点到脖子位置的距离，矩形的宽为长的1/2；以及在原始图像中，确定矩形区域作为目标区域。

根据本公开的实施例，上述识别目标区域中的人像是否以预设方式佩戴安全装置，包括：通过将目标区域的图像输入预设的特征抽取模型，获取目标特征向量；获取正例特征库和负例特征库，正例特征库包括至少一个第一特征向量，负例特征库包括至少一个第二特征向量；通过将目标特征向量分别与正例特征库中的第一特征向量和负例特征库中的第二特征向量进行比对，确定第一距离度量值和第二距离度量值；以及若第一距离度量值小于第二距离度量值，则确定目标区域中的人像以预设方式佩戴安全帽；否则，确定目标区域中的人像没有以预设方式佩戴安全帽。

根据本公开的实施例，上述通过将目标特征向量分别与正例特征库中的第一特征向量和负例特征库中的第二特征向量进行比对，确定第一距离度量值和第二距离度量值，包括：计算目标特征向量与所有第一特征向量之间的欧式距离，并确定目标特征向量与所有第一特征向量之间的欧式距离中最小的欧式距离作为第一距离度量值；以及计算目标特征向量与所有第二特征向量之间的欧式距离，并确定目标特征向量与所有第二特征向量之间的欧式距离中最小的欧式距离作为第二距离度量值。

本公开的另一个方面提供了一种图像识别的装置，包括：第一获取模块，用于获取包含人像的原始图像；第一确定模块，用于确定原始图像中的骨骼信息；第二确定模块，用于根据骨骼信息，从原始图像中确定目标区域；以及识别模块，用于识别目标区域中的人像是否以预设方式佩戴安全装置。

根据本公开的实施例，上述第一确定模块包括：第二获取子模块，用于通过将原始图像输入预设的人体骨骼关键点检测模型，获取骨骼信息，骨骼信息包括脖子位置、左耳位置和右耳位置的坐标。