Vehicle classification for large-scale traffic surveillance videos using Convolutional Neural Networks

被引:46
作者
Zhuo, Li [1 ,2 ]
Jiang, Liying [1 ]
Zhu, Ziqi [1 ]
Li, Jiafeng [1 ]
Zhang, Jing [1 ]
Long, Haixia [1 ]
机构
[1] Beijing Univ Technol, Signal & Informat Proc Lab, Beijing, Peoples R China
[2] Collaborat Innovat Ctr Elect Vehicles Beijing, Beijing, Peoples R China
基金
中国国家自然科学基金; 北京市自然科学基金;
关键词
Vehicle classification; CNN; GoogLeNet; VehicleDataset; Pre-training; Fine-tuning;
D O I
10.1007/s00138-017-0846-2
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Vehicle classification plays an important role in intelligent transport system. However, because the conventional vehicle classification methods are not robust to variations such as illumination, weather, noise, and the classification accuracy cannot meet the requirements of practical applications. Therefore, a new vehicle classification method using Convolutional Neural Networks is proposed in this paper, which consists of two steps: pre-training and fine-tuning. In pre-training, GoogLeNet is pre-trained on ILSVRC-2012 dataset to obtain the initial model with the corresponding connection weights. In fine-tuning, the initial model is further fine-tuned on VehicleDataset which is constructed with 13,700 images in this paper to obtain the final classification model. All images in the VehicleDataset are extracted from real highway surveillance videos, including variations of illumination, noise, resolution, angle of video cameras and weather. The vehicles are divided into six categories, i.e., bus, car, motorcycle, minibus, truck and van. The performance evaluation is carried out on the VehicleDataset. The experimental results show that the proposed method can avoid the complicated process of manually extracting features and the average classification accuracy is up to 98.26%, which is 3.42% higher than the conventional methods using "Feature + Classifier".
引用
收藏
页码:793 / 802
页数:10
相关论文
共 28 条
  • [1] [Anonymous], IEEE
  • [2] [Anonymous], 2006, TECHN INN C 2006 ITI
  • [3] [Anonymous], INT J SMART SENS INT
  • [4] [Anonymous], IEEE T PATTERN ANAL
  • [5] [Anonymous], RES VEHICLE TYPE REC
  • [6] [Anonymous], 2009, TECHNICAL REPORT
  • [7] [Anonymous], NIPS WORKSH
  • [8] [Anonymous], 2014, P 2 INT C LEARNING R
  • [9] [Anonymous], 2013, RES VEHICLE RECOGNIT
  • [10] [Anonymous], 2016, KERAS DEEP LEARNING