[发明专利]一种作业人员安全带识别方法、装置、设备和存储介质

权利要求书

1.一种作业人员安全带识别方法，其特征在于，包括：

获取作业区域图像；

对所述作业区域图像进行预处理，生成预处理图像；

检测所述预处理图像中是否存在目标作业人员；

当所述预处理图像中存在目标作业人员时，从所述预处理图像中确定所述目标作业人员的待识别区域图像；

根据从所述待识别区域图像中提取的方向梯度直方图HOG特征和颜色直方图HOC特征，生成综合特征；

将所述综合特征输入到预置SVM分类模型，输出安全带识别结果；

所述检测所述预处理图像中是否存在目标作业人员的步骤，包括：

将所述预处理图像输入到预置的人员检测模型和预置的梯子检测模型，在所述预处理图像上生成第一图像框和第二图像框；其中，所述第一图像框用于标识人员所在位置，所述第二图像框用于标识梯子所在位置；

当所述第一图像框与所述第二图像框之间存在重叠部分时，计算所述重叠部分的面积与所述第二图像框的面积之间的第一比例；

计算所述重叠部分的底边与所述第二图像框的底边的第一距离；

计算所述第二图像框的顶边和所述第二图像框的底边的第二距离；

确定所述第一距离与所述第二距离之间的第二比例；

当所述第一比例大于或等于第一预设阈值，且所述第二比例大于或等于第二预设阈值时，确定所述预处理图像中存在目标作业人员。

2.根据权利要求1所述的方法，其特征在于，所述对所述作业区域图像进行预处理，生成预处理图像的步骤，包括：

检测所述作业区域图像的多个颜色分量；所述多个颜色分量包括红色分量、绿色分量和蓝色分量；

采用所述红色分量、所述绿色分量和所述蓝色分量，计算所述作业区域图像的灰度值；

根据所述作业区域图像的灰度值构建灰度图像；

对所述灰度图像进行去噪处理，生成预处理图像。

3.根据权利要求1所述的方法，其特征在于，所述当所述预处理图像中存在目标作业人员时，从所述预处理图像中确定所述目标作业人员的待识别区域图像的步骤，包括：

当所述预处理图像中存在目标作业人员时，从所述预处理图像中截取所述第一图像框内的图像作为人员区域图像；

将所述人员区域图像输入到预置的二维姿态估计网络模型，确定所述人员区域图像对应人员的二维关节坐标；

若所述二维关节坐标与预设二维爬高姿态坐标的匹配数量大于第一预设数量阈值，则确定所述人员区域图像对应人员的姿态为爬高姿态；

截取所述人员区域图像的预定比例图像，作为所述目标作业人员的待识别区域图像。

4.根据权利要求3所述的方法，其特征在于，还包括：

将所述二维关节坐标输入到预置的三维姿态估计网络模型，确定所述人员区域图像对应人员的三维关节坐标；

若所述三维关节坐标与预设三维爬高姿态坐标的匹配数量大于第二预设数量阈值，则确定所述人员区域图像对应人员的姿态为爬高姿态；

截取所述人员区域图像的预定比例图像，作为所述目标作业人员的待识别区域图像。

5.根据权利要求1所述的方法，其特征在于，所述根据从所述待识别区域图像中提取的方向梯度直方图HOG特征和颜色直方图HOC特征，生成综合特征的步骤，包括：

去除所述待识别区域图像的背景并进行滤波，生成待处理图像；

将所述待处理图像划分为多个像素小块；

计算所述多个像素小块的梯度方向直方图，得到HOG特征；

在预置的HSV颜色空间中加载所述待处理图像，确定所述待处理图像中各个像素点的色调和饱和度的值；

基于所述各个像素点的色调和饱和度的值构建颜色直方图，得到HOC特征；

对所述HOG特征和所述HOC特征进行归一化处理并进行拼接，生成综合特征。