[发明专利]一种安全帽识别装置及方法

权利要求书

1.一种安全帽识别装置，其特征在于，包括：控制端、无人机和设置在所述无人机上的云台相机；

所述控制端与所述无人机连接；所述无人机与所述云台相机连接；所述无人机用于接收所述云台相机采集到的目标区域的视频图像，并将所述视频图像发送至所述控制端；所述控制端用于输出安全帽识别结果。

2.根据权利要求1所述的安全帽识别装置，其特征在于，还包括：设置在所述无人机上的安全提示模块；

所述安全提示模块与所述无人机连接；所述无人机还用于接收所述控制端发送的安全帽识别结果；所述安全提示模块用于向所述目标区域内未佩戴安全帽的人员发出警示信号。

3.根据权利要求1所述的安全帽识别装置，其特征在于，还包括：设置在所述无人机上的数传模块；

所述数传模块分别与所述控制端和所述无人机连接；所述数传模块用于将所述控制端发出的所述安全帽识别结果发送至所述无人机。

4.根据权利要求1所述的安全帽识别装置，其特征在于，还包括：设置在所述无人机上的图传模块；

所述图传模块分别与所述控制端和所述无人机连接；所述图传模块用于将所述云台相机采集的目标区域的视频图像发送至所述控制端。

5.根据权利要求1所述的安全帽识别装置，其特征在于，所述云台相机为二维相机或三维相机。

6.根据权利要求1所述的安全帽识别装置，其特征在于，所述无人机为四轴固定翼飞行器或六轴固定翼飞行器。

7.根据权利要求4所述的安全帽识别装置，其特征在于，所述图传模块为型号为TS832的图传发射接收器。

8.一种安全帽识别方法，其特征在于，所述方法应用于如权利要求1-7任一项所述的安全帽识别装置，所述方法包括：

获取目标区域的视频图像；所述视频图像是无人机上的云台相机对所述目标区域进行拍摄得到的；

将所述视频图像输入安全帽识别模型中，得到安全帽识别结果；所述安全帽识别模型为采用训练集对YOLOv5网络进行训练得到的。

9.根据权利要求8所述的安全帽识别方法，其特征在于，在所述将所述视频图像输入安全帽识别模型中，得到安全帽识别结果之后，还包括：

将所述安全帽识别结果发送至所述无人机。

10.根据权利要求8所述的安全帽识别方法，其特征在于，所述YOLOv5网络包括空间金字塔池层；所述空间金字塔池层包括依次连接第一卷积层、第一Leaky Relu激活函数、池化层结构、全连接层、第二卷积层和第二Leaky Relu激活函数；所述池化层结构包括五个并行连接的最大池化层。