基于YOLOv8算法的煤矿人员安全帽佩戴检测方法

    Detection method of safety helmet wearing for coal miners based on YOLOv8 algorithm

    • 摘要: 考虑到传统方法存在浪费人力、监督效果不佳等问题,本文研究一种基于安全帽目标图像检测方法的煤矿工人安全帽佩戴检测方法,并使用较新的YOLOv8算法实现安全帽目标的检测。首先构建了基于YOLOv8算法的安全帽目标检测模型。在Ubuntu环境下搭建了基于Pytorch深度模型和CUDA等模块的安全帽目标检测实验环境。采集煤矿井下实际作业场景图像4000张,制作成训练和验证数据集,对构建的基于YOLOv8算法的安全帽目标检测模型进行训练,最终训练出的模型检测准确率达94.4%,满足煤矿井下的安全帽佩戴情况检测需求。

       

      Abstract: Considering the problems of waste of manpower and poor supervision effect in the traditional methods, this paper studies a safety helmet wear detection method for coal miners based on the helmet target image detection method, and uses the newer YOLOv8 algorithm to realize the helmet target detection. Firstly, a helmet target detection model based on YOLOv8 algorithm is constructed. The experimental environment of helmet target detection based on Pytorch depth model and CUDA module is built in Ubuntu environment. 4000 images of actual manufacturers in coal mines are collected and made into training and verification data sets, and the helmet target detection model based on YOLOv8 algorithm is trained. The final training model has a detection accuracy of 94.4%, which meets the detection needs of helmet wearing in coal mines.

       

    /

    返回文章
    返回