一种轻量化YOLOv4的建筑工地安全帽佩戴检测方法及装置

摘要

本发明涉及一种轻量化YOLOv4的建筑工地安全帽佩戴检测方法，属于机器视觉技术领域，包括以下步骤：S1：采集建筑施工现场图像，构建训练和测试数据集；S2：构建轻量化YOLOv4网络用于安全帽佩戴检测；S3：将所述施工现场所有图像经过预处理之后，输入到构建的轻量化YOLOv4网络中进行训练；S4：利用轻量化YOLOv4网络，将训练好的安全帽佩戴检测模型用于实际的建筑工地安全帽佩戴检测。本发明在参数量和计算量上有一定的降低，且能够在背景复杂、小目标及遮挡情况下实现准确检测，以实现建筑工地安全帽佩戴情况的实时检测。