Real-Time Vehicle Color Recognition Based on YOLO9000

Abstract

In this paper, we proposed a real-time automated vehicle color recognition method using you look only once (YOLO)9000 object detection for intelligent transportation system applications in smart city. The workflow in our method contains only one step which achieves recognize vehicle colors from original images. The model proposed is trained and fine tuned for vehicle localization and color recognition so that it can be robust under different conditions (e.g., variations in background and lighting). Targeting a more realistic scenario, we introduce a dataset, called VDCR dataset, which collected on access surveillance. This dataset is comprised up of 5216 original images which include ten common colors of vehicles (white, black, red, blue, gray, golden, brown, green, yellow, and orange). In our proposed dataset, our method achieved the recognition rate of 95.47% and test-time for one image is 74.46 ms.