一种识别未佩戴安全帽的方法、装置及介质.pdf

(19)国家知识产权局(12)发明专利申请(10)申请公布号CN114882535A(43)申请公布日2022.08.09(21)申请号202210635480.3G06V10/82(2022.01)(22)申请日2022.06.07G06K9/62(2022.01)G06N3/04(2006.01)(71)申请人济南博观智能科技有限公司G06N3/08(2006.01)地址250001山东省济南市高新区新泺大街1166号奥盛大厦3号楼17楼(72)发明人沙洁(74)专利代理机构北京集佳知识产权代理有限公司11227专利代理师刘珂(51)Int.Cl.G06V40/10(2022.01)G06V10/26(2022.01)G06V10/44(2022.01)G06V10/74(2022.01)G06V10/764(2022.01)权利要求书2页说明书11页附图4页(54)发明名称一种识别未佩戴安全帽的方法、装置及介质(57)摘要本申请公开了一种识别未佩戴安全帽的方法、装置及介质，涉及图像处理技术领域。包括：提取头肩图的特征；将头肩图的特征输入至预设的分类支路；通过分类支路对头肩图进行是否佩戴安全帽的分类判断；在头肩图未分类为佩戴安全帽的情况下，通过特征匹配支路提取头肩图的特征向量；将提取出的头肩图的向量特征与预先存放在数据库中的样本图的特征向量进行比对；在特征比对的置信度小于阈值时，识别目标未佩戴安全帽。该方法中先在分类支路中判断目标是否未佩戴安全帽，在判断出未佩戴安全帽的情况下，进一步地通过特征匹配支路再次确定目标是否未佩戴安全帽。由此可见，该方法能够减少误识别为未佩戴安全帽的情况，从而提高识别佩戴安全帽的准确性。CN114882535ACN114882535A权利要求书1/2页1.一种识别未佩戴安全帽的方法，其特征在于，包括：获取待测图片中目标的头肩图；其中所述目标的所述头肩图的尺寸小于所述待测图片的尺寸；提取所述头肩图的特征；将所述头肩图的特征输入至预设的分类支路；通过所述分类支路对所述头肩图进行是否佩戴安全帽的分类判断；在所述头肩图未分类为佩戴安全帽的情况下，通过特征匹配支路提取所述头肩图的特征向量；将提取出的所述头肩图的所述特征向量与预先存放在数据库中的样本图的特征向量进行比对；在特征比对的置信度小于阈值的情况下，识别所述目标未佩戴安全帽。2.根据权利要求1所述的识别未佩戴安全帽的方法，其特征在于，所述提取所述头肩图的特征包括：提取所述头肩图的全局特征；通过STN模块对所述头肩图进行校正，并利用强制分割特征模块提取所述头肩图中所述目标的局部特征；拼接所述头肩图的全局特征以及所述头肩图中所述目标的局部特征以作为提取出的所述头肩图的特征。3.根据权利要求2所述的识别未佩戴安全帽的方法，其特征在于，所述通过所述分类支路对所述头肩图进行是否佩戴安全帽的分类判断包括：通过所述分类支路对所述头肩图至少进行戴安全帽、戴除所述安全帽之外的其它帽子、不戴帽子的分类判断。4.根据权利要求1所述的识别未佩戴安全帽的方法，其特征在于，获取预先存放在所述数据库中的所述样本图的特征向量包括：获取至少包括各安全帽样式的图片、在不同方位、不同角度和不同光线强度情况下佩戴各所述安全帽的图片并将获得的各所述图片作为所述样本图；将各所述样本图送入特征提取器以便获取各所述样本图对应的特征向量；将各所述样本图对应的特征向量存放至所述数据库中。5.根据权利要求4所述的识别未佩戴安全帽的方法，其特征在于，在所述特征比对的置信度小于阈值的情况下，识别所述目标未佩戴安全帽之后，还包括：获取误分类为未戴安全帽的所述头肩图；提取误分类为未戴安全帽的所述头肩图的特征；将误分类为未戴安全帽的所述头肩图的特征添加至所述数据库。6.根据权利要求1所述的识别未佩戴安全帽的方法，其特征在于，所述获取待测图片中目标的头肩图，包括：使用Yolov5检测所述待测图片中所述目标的头肩位置坐标；从所述待测图片中抠出所述目标的所述头肩图。7.根据权利要求1至6任意一项所述的识别未佩戴安全帽的方法，其特征在于，在所述特征比对的置信度小于阈值的情况下，识别所述目标未佩戴安全帽之后，还包括：2CN114882535A权利要求书2/2页输出用于提示未戴安全帽的提示信息。8.一种识别未佩戴安全帽的装置，其特征在于，包括：获取模块，用于获取待测图片中目标的头肩图；其中所述目标的所述头肩图的尺寸小于所述待测图片的尺寸；第一提取模块，用于提取所述头肩图的特征；输入模块，用于将所述头肩图的特征输入至预设的分类支路；判断模块，用于通过所述分类支路对所述头肩图进行是否佩戴安全帽的分类判断，若否，则触发第二提取模块；所述第二提取模块，用于在所述头肩图未分类为佩戴安全帽的情况下，通过特征匹配支路提取所述头肩图的特征向量