基于多模块优化及Slim轻量化的道路凹陷检测

Multi-module optimization and slim lightweight road depression detection

  • 摘要: 为提高道路凹陷检测的精度并降低模型复杂度,提出了一种轻量化目标检测算法PSA-YOLO. 首先,在特征提取阶段引入AKConv模块,通过动态调整卷积核优化多尺度特征表示,并结合ADown下采样模块增强特征图空间维度的关联性;然后,在特征融合阶段,设计了StarECA-C2f模块,采用自适应通道权重分配机制,有效提升细微特征的捕捉能力,在检测阶段设计了轻量级检测头LSCH,利用共享卷积和可学习尺度层减少计算负担,并结合WIoUv3多方向权重调整机制,有效减少精度损失;最后,通过Slim剪枝方法去除冗余网络通道,进一步简化模型结构,降低计算和存储开销. 实验结果表明,与YOLOv8n相比,PSA-YOLO在参数量和计算量上分别减少了57%和50%,同时mAP50和mAP50-95分别提升了2.0%和1.9%;在GRDDC2020数据集上测试时,PSA-YOLO的mAP50和mAP50-95分别提升了0.6%和0.3%,显示了该模型在不同数据集上良好的性能和适应能力.

     

    Abstract: In order to improve the accuracy and reduce the model complexity of road depression detection, a lightweight target detection algorithm PSA-YOLO is proposed.Firstly, the AKConv module is introduced in the feature extraction stage, which optimizes the multi-scale feature representation by dynamically adjusting the convolution kernel and enhances the correlation of the spatial dimensions of the feature maps by combining with the ADown downsampling module.Then, in the feature fusion stage, the design of the StarECA-C2f module with adaptive channel weight assignment mechanism to effectively enhance the capture ability of fine features, and in the detection stage, a lightweight detection head LSCH is designed to reduce the computational burden by utilizing shared convolution and learnable scale layers, and combined with the WIoUv3 multi-directional weight adjustment mechanism to effectively reduce the loss of accuracy. Finally, redundant network channels are removed by Slim pruning method to further simplify the model structure and reduce the computational and storage overhead. The experimental results show that compared with YOLOv8n, PSA-YOLO reduces the number of parameters and computation by 57% and 50%, while mAP50 and mAP50-95 are improved by 2.0% and 1.9%, respectively; when tested on the GRDDC2020 dataset, PSA-YOLO's mAP50 and mAP50-95 are improved by 0.6% and 0.3%, showing the good performance and adaptability of the model on different datasets.

     

/

返回文章
返回