长安大学学报(自然科学版)2020,Vol.40Issue(4) :109-116.DOI:10.19721/j.cnki.1671-8879.2020.04.011

基于SqueezeNet卷积神经网络的车辆颜色识别

Vehicle color recognition based on SqueezeNet

李坤伦 魏泽发 宋焕生
长安大学学报(自然科学版)2020,Vol.40Issue(4) :109-116.DOI:10.19721/j.cnki.1671-8879.2020.04.011

基于SqueezeNet卷积神经网络的车辆颜色识别

Vehicle color recognition based on SqueezeNet

李坤伦 1魏泽发 2宋焕生3
扫码查看

作者信息

  • 1. 长安大学 教育技术与网络中心,陕西西安710064;长安大学电子与控制工程学院,陕西西安710064
  • 2. 长安大学 教育技术与网络中心,陕西西安710064
  • 3. 长安大学 教育技术与网络中心,陕西西安710064;长安大学信息工程学院,陕西西安710064
  • 折叠

摘要

为实现车辆颜色的精准识别,辅助现有车牌识别方法在智能交通系统中完成对车辆身份的精确认证,通过分析现有解决方案和探究卷积神经网络在实际应用中的问题,提出基于轻量级卷积神经网络SqueezeNet实现对车辆颜色识别的方法.轻量级卷积神经网络SqueezeNet的参数量是AlexNet网络结构的1/50,同时还能保证精度,避免由于网络结构复杂、参数量大造成的大规模计算和较高的计算机硬件需求,提升了模型的可移植性.选取车辆颜色识别(Vehicle_Color_Recognition)数据集作为研究基础,针对数据集进行了扩充和增强处理.以SqueezeNet为基准探究了特征融合对分类结果的影响,通过对比试验确定将fire7/concat输出特征图和fire9/concat输出特征图相融合.研究结果表明:轻量级卷积神经网络SqueezeNet在保证模型大小2.9M、单张测试时间15 ms的基础上,训练精度为96.28%,而AlexNet的模型大小为227M、单张测试时间24 ms、训练精度为96.18%;在实现同等精度的前提下,轻量级卷积神经网络SqueezeNet训练得到的模型更适合移植到如现场可编程门列阵(FPGA)这种开发板上,同时在服务器上的处理速度也更快;融合后的模型最终的分类结果提升为96.48%.利用轻量级卷积神经网络SqueezeNet进行车辆颜色识别可以较好地应用在智能交通系统中,并在一定程度上解决目前车牌识别的难点.

关键词

交通工程/卷积神经网络/智能交通/车辆颜色识别/SqueezeNet

引用本文复制引用

基金项目

国家自然科学基金(61572083)

教育部联合基金(6141A02022610)

出版年

2020
长安大学学报(自然科学版)
长安大学

长安大学学报(自然科学版)

CSTPCDCSCD北大核心
影响因子:1.011
ISSN:1671-8879
被引量11
参考文献量1
段落导航相关论文