一种基于YOLO网络压缩算法的安全帽识别智能监控系统

摘要

本发明公开了一种基于YOLO网络压缩算法的安全帽识别智能监控系统，属于计算机视觉和数字图像处理技术领域，本发明提供的监控系统包括获取训练数据的数据处理模块、设置及网络结构及训练网络参数的特征提取与预测模块、对所设置的网络模型的全连接层进行压缩处理的网络压缩模块和结果检测模块，本发明采用YOLO网络，降低背景误检率低，同时可以直接优化性能检测，提高准确率；以及直接对锚框进行回归和分类，加快了运行速度，实现了实时处理视频，以及可以学到物体更泛化的特征表示，更能适应新的领域，泛化能力高，可高度可推广。本发明能用于对施工区域工作人员是否佩戴安全帽的智能监控。