• 复杂施工现场环境下的轻量化安全检测算法

    Lightweight construction site safety detection algorithm in complex environments

    • 为解决现有施工安全检测模型在复杂环境施工现场中安全检测精度低、难以在性能较低的移动端设备中部署的问题,提出了一种改进YOLOv8n施工安全检测模型CMSFD-YOLO。首先,在网络主干中引入通道先验卷积注意力机制(Channel Prior Convolutional Attention, CPCA),增强模型在复杂环境下对施工安全特征的提取能力。随后,结合多尺度序列特征融合模块(Multi Scale sequence Feature fusion, MSF)和三重特征编码模块(Triple Feature Encoder, TFE),构建CMSFD特征金字塔,充分结合多类施工安全目标信息,优化多尺度目标特征融合效率。其次,设计轻量级共享细节增强卷积(Lightweight Shared Detail-Enhanced Convolutional Detection, LSDECD)检测头,对模型进行轻量化,减小参数量、计算量和网络模型体积,同时平衡检测精度。最后,引入Focaler-EIoU作为边界框回归的损失函数,优化模型收敛速度和回归结果。在自建的SODA_VOC数据集上进行实验,CMSFD-YOLO模型的mAP@0.5达到86.2%,相较于YOLOv8n提升了4.3%,而参数量、计算量和模型尺寸方面分别只有2266728、6.5×109和4.39 MB,相较于YOLOv8n分别降低了24.6%、19.7%和26.2%。最终将改进模型部署到移动端设备中,进行实际效果检测,证明了改进后模型在复杂施工现场环境下的安全检测任务中具有良好的应用性能。

       

      Abstract: In order to solve the problems of low safety detection accuracy of existing construction safety detection models in complex environment construction sites and difficult to be deployed in low performance mobile devices, this paper proposes an improved YOLOv8n construction safety detection model CMSFD-YOLO.First, a Channel Prior Convolutional Attention (CPCA) mechanism is introduced into the network backbone to enhance the model's ability to extract construction safety features in complex environments. Convolutional Attention (CPCA) is introduced into the network backbone to enhance the model's ability to extract construction safety features in complex environments. Subsequently, the CMSFD feature pyramid is constructed by combining the Multi Scale sequence Feature fusion (MSF) module and the Triple Feature Encoder (TFE) module, which fully combines the information of multiple types of construction safety targets to optimize the efficiency of multi-scale target feature fusion. Second, the Lightweight Shared Detail-Enhanced Convolutional Detection (LSDECD) detection head is designed to lighten the model and reduce the number of parameters, computation and network model volume, while balancing the detection accuracy. Finally, Focaler-EIoU is introduced as the loss function for bounding box regression to optimize the model convergence speed and regression results. Experiments on the self-built SODA_VOC dataset, and the CMSFD-YOLO model mAP@0.5 it reached 86.2%, are improved by 4.3% compared to YOLOv8n, while the number of parameters, computational volume and model size are only 2266728, 6.5×109 and 4.39 MB, respectively, which are reduced by 24.6% compared to YOLOv8n, 19.7% and 26.2%, respectively, compared to YOLOv8n. Finally, the improved model is deployed to mobile devices for real-world effect testing, which proves that the improved model has good application performance in safety detection tasks in complex construction site environments.

       

    /

    返回文章
    返回