Deep Learning-Based Vehicle Classification Using an Ensemble of Local Expert and Global Networks

Jong Taek Lee, Yunsu Chung; Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops, 2017, pp. 47-52

Vehicle classification has been a challenging problem because of pose variations, weather / illumination changes, inter-class similarity and insufficient training dataset. With the help of innovative deep learning algorithms and large scale traffic surveillance dataset, we are able to achieve high performance on vehicle classification. In order to improve performance, we propose an ensemble of global networks and mixture of K local expert networks. It achieved a mean accuracy of 97.92%, a mean precision of 92.98%, a mean recall of 90.24% and a Cohen Kappa score of 96.75% on unseen test dataset from the MIO-TCD classification challenge.