• 基于改进YOLOv7-Tiny的佩戴安全帽检测算法

    A wearing safety helmet detection algorithm based on the improved YOLOv7-Tiny

    • 施工现场的监控一般安装在高处,这导致利用摄像头采集的工人佩戴安全帽的数据存在大量的被遮挡目标和小目标。这些目标像素少且特征易与背景混淆,会出现漏检和误检等问题。为了解决这些问题,提出了CAH-YOLO算法。首先,在YOLOv7-Tiny的主干中引入C-ELAN(Contextual Efficient Layer Aggregation Network)网络,通过大核卷积增强目标的上下文特征提取。其次,在YOLOv7-Tiny的颈部中添加ASCF(Adaptive Feature Space Channel Fusion)网络,通过充分融合不同层的信息增强有用信息的表达。最后,YOLOv7-Tiny中的下采样模块替换为HWD(Haar Wavelet-based Downsampling)并添加了HWD组模块,在降低参数量的同时保留更多的有用信息。实验显示,CAH-YOLO算法的平均精度均值指标mAP@0.5提高了2%、mAP@0.75提高了4.5%,对小目标的指标mAP@s提高了3.2%。

       

      Abstract: The surveillance cameras on the construction site are generally installed at high positions, which leads to a large number of obscured and small objects in the collected data wearing safety helmets. These objects have fewer pixels and their features are easily confused with the background, which is prone to miss and false detection. To solve these problems, this paper proposed the CAH-YOLO algorithm. Firstly, the Contextual Efficient Layer Aggregation Network (C-ELAN) is introduced into the backbone module of YOLOv7-Tiny, utilizing large kernel convolutions to enhance contextual feature extraction for objects. Secondly, the Adaptive Space Channel Fusion (ASCF) network is added to the neck module of YOLOv7-Tiny, enhancing the expression of useful information by fully integrating information from different layers. Finally, the downsampling modules of YOLOv7-Tiny are replaced with the Haar Wavelet-based Downsampling (HWD) modules and a HWD group module are added, which reduces the number of parameters while retaining more useful information. Experiments show that the CAH-YOLO algorithm improves the mAP@0.5 by 2%, mAP@0.75 by 4.5% and mAP@s for small objects by 3.2%.

       

    /

    返回文章
    返回