README.md

Welcome color-recognition repo

This model can only recognise the following colours

1. Red
2. Blue
3. Orange
4. Green
5. Yellow
6. Indigo
7. Voilet

Problem Statement

Interns are expected to create a TensorFlow model for identification of color. 
Your code will be given an image as an input and the code is supposed to categorize 
which RGB clour it matches to. This code is to be done in Python.

Solution

For implemention of this project we need dataset.

So color dataset I have created using numpy and RGB color codes

Code for Dataset genrator is located in following directory

    color-recognition/Dataset/DatasetGenerator/

Generation dataset and Model training

To Generate dataset type following command

python Generate.py

this ll generate dataset and stored in test dir which is located in DataGenerator.

Using generated dataset I have trained a keras-tf model which is located in src/keras_model.h5

keras_model.h5 is the trained model by tensorflow.

Requirement for this model are

python3.X
tensorflow
pillow library
matplotlib library

Use

To use this model we need to put input image file on src folder

and in `code.py` give input of file name to `line no 17`

Run file


`python code.py`

Thank you

All done !!

Predicted output will be shown to you