重庆理工大学学报(自然科学) ›› 2020, Vol. 34 ›› Issue (10): 160-168.doi: 10.3969/j.issn.1674-8425(z).2020.10.022

• 电气·电子 • 上一篇    下一篇

一种面向多视角交通标志识别的CapsNet集成算法

屈治华,王 琳,邵毅明,邓天民   

  1. 重庆交通大学交通运输学院,重庆 400074
  • 收稿日期:2019-09-22 发布日期:2020-11-16
  • 作者简介:屈治华,男,博士研究生,主要从事智能驾驶、计算机视觉方面研究,Email:71337188@qq.cpm。
  • 基金资助:
    国家重点研发计划项目(2016YFB0100905);重庆市科技人才培养计划项目(cstc2013kjrc-qnrc0148)

  • Received:2019-09-22 Published:2020-11-16

摘要: 为解决传统卷积神经网络在交通标志识别过程中因最大池化层的固有缺陷导致无法对图像的姿态、位置等实例化参数进行检测的问题,提出了一种基于Bagging集成的CapsNet交通标志识别算法。以CapsNet作为基分类器构建Bagging集成框架,设定可自适应过采样倍率的SMOTE算法对少数类样本进行过采样,提升算法整体识别精度。在同等测试环境下选取多种算法,针对德国交通标志数据库(GTSRB)和经过仿射变换后的多视角数据集的识别精度、耗时和泛化能力进行对比。实验结果表明:所提出算法的识别精度为99.07%,且在偏转角度达到30°时,准确率仍能保持77.58%,可有效提高多视角下交通标志的识别精度。

关键词: 智能交通;交通标志识别;CapsNet;集成算法;多视角图像识别

中图分类号: 

  • TP391.41