基于改进YOLOv4-Tiny的交通标志图像识别算法研究

TRAFFIC SIGN IMAGE RECOGNITION ALGORITHM BASED ON IMPROVED YOLOv4-TINY

  • 摘要: 为实现无人驾驶汽车对交通标志的精准识别,提出基于改进YOLOv4-Tiny的交通标志图像识别算法YOLO-slim。在原算法中加入卷积注意力网络并在特征金字塔网络中引入浅层特征,提高算法对不同层间特征信息的利用率。使用深度可分离卷积替换标准卷积减少网络参数量压缩模型权重文件。在模型训练中使用Focus loss损失函数平衡难易样本。实验结果表明,YOLO-slim的平均准确率为94.41%,权重文件为4.49 MB,检测速度为8.0 ms。改进后的算法准确率更高、权重文件更小,更适合部署在车载计算单元上。

     

    Abstract: In order to realize the accurate recognition of traffic signs by autonomous vehicle, a traffic sign image recognition algorithm YOLO-slim based on improved YOLOv4-Tiny is proposed. Convolution attention module was added to the original network and shallow features were introduced into the feature pyramid network to improve the utilization rate of feature information between different layers. Depthwise separable convolution was used to replace standard convolution to reduce the number of network parameters and compress the model weight file. Focus loss function was used to balance difficult samples in model training. Experimental results show that YOLO-slim's mean average precision is 94.41%, weight file is 4.49 MB, and detection speed is 8.0 ms. The improved algorithm has higher accuracy and smaller weight files, and is more suitable for deployment in vehicle-mounted computing units.

     

/

返回文章
返回