[发明专利]视频中人员安全帽佩戴状态的检测方法、装置及电子设备

权利要求书

1.一种视频中人员安全帽佩戴状态的检测方法，其特征在于，包括：

获取当前视频中包含目标对象的实时图像；

从所述实时图像中确定所述目标对象的关键骨骼点；

根据所述目标对象的关键骨骼点，在所述实时图像中确定对应所述目标对象的头部检测区域；

利用预设的安全帽检测模型，判断所述头部检测区域是否存在安全帽特征；

若所述头部检测区域存在安全帽特征，则确定所述目标对象处于安全帽佩戴状态。

2.根据权利要求1所述的方法，其特征在于，所述从所述实时图像中确定所述目标对象的关键骨骼点的步骤，包括：

利用人体检测算法，从所述实时图像中确定对应所述目标对象的人体检测框；

利用所述人体检测框依次预测所述目标对象的姿态，得到所述目标对象的多个备用骨骼点；

从所确定的多个备用骨骼点中，筛选出来五个关键骨骼点，五个所述关键骨骼点包括左侧肩部骨骼点、右侧肩部骨骼点、中央颈部骨骼点、左侧眼眶骨骼点和右侧眼眶骨骼点。

3.根据权利要求2所述的方法，其特征在于，所述根据所述目标对象的关键骨骼点，在所述实时图像中确定对应所述目标对象的头部检测区域的步骤，包括：

以所述左侧肩部骨骼点、右侧肩部骨骼点和中央颈部骨骼点为基准绘制底部边缘线；

以所述左侧眼眶骨骼点和右侧眼眶骨骼点为基准绘制中线；

以所述中线为基准绘制与所述底部边缘线对称的顶部边缘线；

根据所述顶部边缘线和所述底部边缘线绘制矩形的所述头部检测区域。

4.根据权利要求3所述的方法，其特征在于，所述以所述中线为基准绘制与所述底部边缘线对称的顶部边缘线的步骤之后，所述方法还包括：

判断所述顶部边缘线是否低于或者等于所述实时图像的顶部边界线；

若所述顶部边缘线低于或者等于所述实时图像的顶部边界线，保持所述顶部边缘线不变；

若所述顶部边缘线高于所述实施图像的顶部边界线，以所述实时图像的顶部边界线作为所述顶部边缘线。

5.根据权利要求1至4中任一项所述的方法，其特征在于，所述利用预设的安全帽检测模型，判断所述头部检测区域是否存在安全帽特征的步骤之后，所述方法还包括：

若所述头部检测区域不存在安全帽特征，则确定所述目标对象处于安全帽未佩戴状态，发起语音报警。

6.根据权利要求5所述的方法，其特征在于，所述利用预设的安全帽检测模型，判断所述头部检测区域是否存在安全帽特征的步骤之前，所述方法还包括：

获取预设数量的样本图像，其中，所述样本图像为包括头部佩戴安全帽特征的图像和头部未佩戴安全帽特征的图像；

利用全部所述样本图像训练神经网络模型，得到能够检测图像中头部安全帽佩戴状态的二分类模型，作为所述安全帽检测模型。

7.根据权利要求6所述的方法，其特征在于，所述神经网络模型为Alex Net、Res Net、Mobile Net、VGG中的任一种。

8.一种视频中人员安全帽佩戴状态的检测装置，其特征在于，包括：

获取模块，用于获取当前视频中包含目标对象的实时图像；

第一确定模块，用于从所述实时图像中确定所述目标对象的关键骨骼点；

第二确定模块，用于根据所述目标对象的关键骨骼点，在所述实时图像中确定对应所述目标对象的头部检测区域；

判断模块，用于利用预设的安全帽检测模型，判断所述头部检测区域是否存在安全帽特征；

第三确定模块，用于若所述头部检测区域存在安全帽特征，则确定所述目标对象处于安全帽佩戴状态。