融合注意力机制的轻量化行人检测算法

LIGHTWEIGHT PEDESTRIAN DETECTION ALGORITHM BASED ON ATTENTION MECHANISM

  • 摘要: 针对交通场景行人检测模型参数多、检测精度较低的问题,提出一种融合注意力机制的轻量化行人检测算法。借鉴 Ghost 思想对 YOLOv5 进行模型轻量化处理;在数据处理部分融合图像混合增强算法,并在特征提取网络中嵌入坐标注意力 (Coordinate Attention, CA),提高行人检测精度;改进回归优化损失函数,提升训练速度和推理准确性。将改进算法在处理后的 Caltech 行人数据集上进行实验,结果表明,改进算法平均检测精度 (IOU=0.5) 81.7%,较 YOLOv5 提高 4.4 百分点,且模型参数量降低 45.1%,仅为 3.9×10⁶。

     

    Abstract: Aimed at the problem of many parameters and low detection accuracy of pedestrian detection model in traffic scene, a lightweight pedestrian detection algorithm integrating attention mechanism is proposed. The model lightweight processing was carried out on YOLOv5 with reference to the Ghost idea. The image hybrid enhancement algorithm was integrated in the data processing part, and coordinate attention (CA) was embedded in the feature extraction network to improve the pedestrian detection accuracy. The regression optimization loss was improved function to improve training speed and inference accuracy. The improved algorithm was tested on the processed Caltech pedestrian dataset. The results show that the average detection accuracy (IOU=0.5) of the improved algorithm is 81.7%, which is 4.4 percentage points higher than that of YOLOv5, and the model parameters are reduced by 45.1%, 3.9×10⁶.

     

/

返回文章
返回