[发明专利]基于关键点识别变电站人员安全帽佩戴的检测方法

权利要求书

1.一种基于关键点识别变电站人员安全帽佩戴的检测方法，其特征在于，包括：

1)建立变电站人员的安全帽佩戴标注数据集，并对所述标注数据集送入目标检测模型：SSD网络目标检测模型；

2)对待检测安全帽佩戴的人员的图像，利用训练好的AlphaPose模型进行人体关键点识别；

3)对生成的人体关键点进行三角定位确定头部位置，采用训练好的SSD网络目标检测模型进行安全帽佩戴检测；

其中，所述三角定位确定头部位置的方法包括：

3-1)根据得到人体关键点，获得关键部位的位置信息，即根据鼻子、左肩、右肩三点的位置信息进行计算，得到检测的头部区域，具体包括：

(3-1-1)通过鼻子、左肩、右肩三点位置确定一个三角形区域；设所述鼻子、左肩、右肩的位置分别为：

H(x₀,y₀),Ls(x₁,y₁),Rs(x₂,y₂) (I)

在公式(I)中，x₀，y₀为鼻子所在图像内的横纵坐标；x₁,y₁为左肩所在图像内的横纵坐标；x₂,y₂为右肩所在图像内的横纵坐标；

(3-1-2)以鼻子为中心将所述三角形区域旋转180°得到新三角形，设旋转后左肩和右肩的位置分别为：

Ls′(x₁′,y₁′),Rs′(x₂′,y₂′) (II)

在公式(II)中，x₁′,y₁′为左肩旋转后所在图像内的横纵坐标；x₂′,y₂′为右肩旋转后所在图像内的横纵坐标；

旋转后左肩和右肩的位置为：

(Ls′(x₁′,y₁′),Rs′(x₂′,y₂′))＝(Ls′(2x₀-x₁，2y₀-y₁),Rs′(2x₀-x₂,2y₀-y₂)) (III)

(3-1-3)根据步骤(3-1-1)中得到的原三角形和步骤(3-1-2)中旋转得到的新三角形共同组成一个四边形，则组成四边形的对角线长度分别为：

(3-1-4)以步骤(3-1-3)所述四边形的最长对角线为直径，对角线的交点为圆心作最大外圆，再求得所述外圆的外接正方形，所述外接正方形所框选的区域即为步骤(3-1)中的头部区域；

3-2)对待检测安全帽佩戴的人员的图像进行检测，包括：

(3-2-1)对步骤(3-1-4)选定检测区域，即头部区域，利用训练好的AlphaPose模型作为目标检测模型，进行安全帽佩戴检测；

(3-2-2)根据步骤(3-2-1)得到安全帽佩戴状态的检测结果：

若人员佩戴安全帽，则继续检测；

若人员未佩戴安全帽，则给出反馈结果。

2.根据权利要求1所述一种基于关键点识别变电站人员安全帽佩戴的检测方法，其特征在于，所述反馈结果包括但不限于：人员未佩戴安全帽的纯文本信息或人员未佩戴安全帽的框选标识。

3.根据权利要求1所述一种基于关键点识别变电站人员安全帽佩戴的检测方法，其特征在于，所述步骤1)建立变电站人员的安全帽佩戴标注数据集的步骤包括：

1-1)通过对变电站视频流截取的方式获得变电站人员的图片；

1-2)采集步骤1-1)图片对应各个作业地点中单人以及多人的各种姿态的图像；

1-3)对步骤1-2)中的图像进行扩增，得到待标注的图像数据集。