Vehicle type classification and attribute prediction using multi-task RCNN

Zhuoqun Huo, Yizhang Xia, Bailing Zhang

Research output: Chapter in Book or Report/Conference proceedingConference Proceedingpeer-review

37 Citations (Scopus)

Abstract

Vehicle classification is an important subject of study due to its significance in a number of areas including law enforcement, traffic surveillance, autonomous navigation, and transportation management. While numerous approaches have been proposed, few studies have been published with regard to the multi-view classification of vehicles captured in real surveillance. In this paper, we consider the multi-view classification of vehicles as an attribute prediction problem with views (rear, front, and side) as attributes. The corresponding multi-task learning is implemented in the Region-based Convolutional Neural Network (RCNN) framework, which classifies vehicle categories (car, truck, bus, and van) and predicts the attributes simultaneously. Experiments on a field-captured vehicle dataset provide satisfactory results, with approximate 83% accuracy for vehicle type classification and over 90% accuracy for attribute prediction.

Original languageEnglish
Title of host publicationProceedings - 2016 9th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics, CISP-BMEI 2016
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages564-569
Number of pages6
ISBN (Electronic)9781509037100
DOIs
Publication statusPublished - 13 Feb 2017
Event9th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics, CISP-BMEI 2016 - Datong, China
Duration: 15 Oct 201617 Oct 2016

Publication series

NameProceedings - 2016 9th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics, CISP-BMEI 2016

Conference

Conference9th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics, CISP-BMEI 2016
Country/TerritoryChina
CityDatong
Period15/10/1617/10/16

Keywords

  • Vehicle type recognition
  • attribute prediction
  • convolutional neural networks
  • multi-task learning

Fingerprint

Dive into the research topics of 'Vehicle type classification and attribute prediction using multi-task RCNN'. Together they form a unique fingerprint.

Cite this