一种安全帽佩戴检测方法、装置、设备以及存储介质
摘要:
本发明公开了一种安全帽佩戴检测方法、装置、设备以及计算机可读存储介质,包括:将待检测图像输入至目标安全帽检测神经网络模型,利用目标检测SSD网络提取人体目标特征图;将人体目标特征图输入至关键区域注意力机制网络确定头部关键区域;根据头部关键区域的位置,利用关键区域特征图分割网络从人体目标特征图中分割提取头部区域特征图;将头部区域特征图输入至安全帽检测动态模型,对分支网络输出的置信度进行判定,输出置信度大于等于阈值的检测结果作为安全帽佩戴检测结果。本发明所提供的方法、装置、设备以及计算机可读存储,提高了安全帽佩戴情况检测结果的精度与效率,有效避免了针对变电站场景的复杂背景出现的漏检与误检的现象。
0/0