[发明专利]佩戴安全帽识别系统

权利要求书

1.佩戴安全帽识别系统，包括以下模块：

第一个是视频处理模块，将视频流切成图片；

第二个是检测模块，将人头检测出来；

第三个是分类模块，将佩戴安全帽和不佩戴安全帽分类出来。

所述的佩戴安全帽识别系统的工作流程为：获取到现场的视频流，将视频流按帧切成图片，使用目标检测方法将人头图片截取下来，最后送入到分类网络中对图像进行分类。

2.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的视频处理模块对于视频流的处理使用开源的技术FFmpeg将视频流按照一秒一帧的方式截取出图片。

3.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的视频处理模块首先通过现场的摄像头得到视频，再使用FFmpeg工具将视频分解成图片，之后输入到所述的检测模块中。

4.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的检测模块采用的yolov5检测算法。

5.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的检测模块主要检测部分为人头部分。

6.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的检测模块从所述的视频处理模块得到图片，将检测到的人头输入到下一个分类模块中。

7.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的检测模块利用内部采集的数据作为训练数据。

8.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的分类模块采用的算法是ResNet50。

9.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的分类模块将检测对象的类别分成三类，分别是类别一佩戴安全帽，类别二不佩戴安全帽，类别三未知情况。

10.如权利要求1所述的佩戴安全帽识别系统，其特征在于所述的每个模块之间都有详细目的，同时各个模块之间共同工作。