液晶与显示, 2019, 34 (12): 1191, 网络出版: 2020-01-09   

基于深度可分离卷积的交通标志识别算法

Traffic sign recognition algorithm based on depthwise separable convolutions
作者单位
天津大学 微电子学院, 天津 030072
摘要
针对目前实时交通标志识别中出现的对于中小型目标检测精度低的问题, 本文提出了一种基于YOLOv3-tiny的深度可分离卷积(Depthwise Separable Convolutions)的轻量级交通标志检测网络。采用深度可分离卷积构建深度可分离卷积模块代替普通卷积搭建特征提取网络,在保证计算量的前提下更好地提取中小型目标的特征信息。同时, 改进多尺度特征融合网络, 提高对中小型交通标志的检测精度, 使用h-swish激活函数减少因为网络层数增加而丧失的图像特征, 实现对多类交通标志的检测。实验结果表明: 本算法有效的提高了对中小型交通标志的检测, 在验证集上对警告标志(Warining)指示标志(Mandatory)、禁止标志(Prohibitory)3类交通标志进行检测, 检测精度(AP)结果分别为9857%, 96.03%, 98.04%。检测平均精度(mAP)97.54%、检测速度为201.5 f/s.平均精度较YOLOv3-tiny提高了1401%。在保证轻型网络的计算量低、检测时效性好的前提下, 有效地提升了检测精度。
Abstract
In order to solve the problem of low precision detection for small and medium-sized targets in the real-time traffic sign recognition, a lightweight traffic sign detection network based on YOLOv3-tiny depthwise separable convolution is proposed. In this paper, a deep separable convolution module is constructed by using deep separable convolution instead of the common convolution to construction feature extraction network,the feature information of small and medium-sized targets is better extracted under the premise of ensuring the calculation amount.At the same time, the multi-scale feature fusion network is improved to improve the detection accuracy of small and medium-sized traffic signs. The h-swish activation function is used to reduce the image features lost due to the increase of the number of network layers, and the detection of multiple types of traffic signs is realized.The experimental results show that the algorithm effectively improves the detection of small and medium-sized traffic signs. Warining, mandatory and prohibitory traffic signs are detected on the verification set. The detection accuracy (AP) is 98.57%, 96.03% and 98.04% respectively. The average detection accuracy (mAP) was 97.54% and the detection speed was 201.5 f/s. The average accuracy was 14.01% higher than that of YOLOv3-tiny. The algorithm effectively improves the detection accuracy under the premise of ensuring low calculation of light network and good timeliness of detection.

杨晋生, 杨雁南, 李天骄. 基于深度可分离卷积的交通标志识别算法[J]. 液晶与显示, 2019, 34(12): 1191. YANG Jin-sheng, YANG Yan-nan, LI Tian-jiao. Traffic sign recognition algorithm based on depthwise separable convolutions[J]. Chinese Journal of Liquid Crystals and Displays, 2019, 34(12): 1191.

本文已被 2 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!