基于迁移学习的交通标志识别系统设计 |
| |
引用本文: | 刘玉民,张雨虹.基于迁移学习的交通标志识别系统设计[J].唐山学院学报,2023(6):1-4. |
| |
作者姓名: | 刘玉民 张雨虹 |
| |
作者单位: | 1. 唐山学院机电工程学院;2. 唐山学院计算中心 |
| |
摘 要: | 针对自动驾驶领域对交通标志识别的需求,设计了一种基于迁移学习的交通标志识别系统。该系统采用了预训练的MobileNetV3(去掉输出层)作为特征提取网络,然后添加两个自定义的全连接层以实现信号的分类和输出。由于采用迁移学习方法,深度学习网络中需训练的模型参数大幅减少,训练所需时间大为缩短。该系统使用经典的中国交通标志数据库(CTSDB)中的数据作为交通标志的训练数据和测试数据,训练结果表明,损耗低至0.024 3,准确率高达99.88%;测试结果表明,可以对58类交通标志进行识别,准确率为55.3%。
|
关 键 词: | 迁移学习 交通标志 识别系统 自动驾驶 |
|
|