/Color-Classification-CNN

color recognition (convolutional neural network implemented in Keras)

Primary LanguageJupyter Notebook

Color-Classification-CNN

Color recognition implemented in Keras. This is an Keras implementation of CNN. The implementation supports TensorFlow backends.

The objective of the original model is to detect vehicle colors from traffic cameras. In the paper, they presented the vehicle color recognition system using CNN.The model successfully captured vehicle color in very high accuracy, 94.47%. To know more about how the model works, please refer to the original paper (Vehicle Color Recognition Using Convolutional Neural Network , Reza Fuad Rachmadi and I Ketut Eddy Purnama)

Model Architecture

architecture

Confusion Matrix of the model

architecture