基于YOLOv8n改进的井下人员安全帽佩戴检测
Improved helmet wearing detection of underground personnel based on YOLOv8n
-
摘要: 针对现有井下人员安全帽佩戴检测算法具有漏检误检、精度差以及模型不够轻量化等问题,提出了基于YOLOv8n改进的井下人员安全帽佩戴检测算法。针对井下安全帽多为小目标的情况,加入P2检测层以及检测头;再引入CBAM注意力机制对图像进行关键特征的提取;进一步将Wise-IoU替换CIoU损失函数,使得模型训练效果得到提升;最后将检测头替换为LSCD使得模型轻量化。在开源数据集DsLMF+中的矿山安全帽上的实验结果证明:该方法的最终识别率上升了1.8个百分点达到了94.8%,参数量(parameters)降低了23.8%,计算量(GFLOPs)降低了 10.4%,模型大小(Module size)降低了17.2%,能够实现在井下对人员安全帽佩戴实时并准确的检测。Abstract: In order to solve the problems of missed detection, poor accuracy and insufficient lightweight model of the existing safety helmet wearing detection algorithm for underground personnel, an improved safety helmet wearing detection algorithm for underground personnel based on YOLOv8n was proposed. In view of the situation that the downhole helmet is small target, the P2 detection layer and the detection head are added, the CBAM attention mechanism is introduced to extract the key features of the image, the Wise-IoU is further introduced to replace the CIoU loss function to improve the training effect of the model, and finally the detection head is replaced by LSCD to make the model lightweight. The experimental results on the mine helmet in the open-source dataset DsLMF+ show that the final recognition rate of the method increases by 1.8 percentage points to 94.8%, the number of parameters is reduced by 23.8%, the amount of computation (GFLOPs) is reduced by 10.4%, and the model size is reduced by 17.2%, which can realize real-time and accurate detection of personnel wearing safety helmets underground.
点击查看大图
计量
- 文章访问数: 46
- HTML全文浏览量: 14
- 被引次数: 0