parikhv15 / image-recognition-tensorflow

Image Recognition

Setup

1. First install all the required dependencies using the command below:

pip install -r requirements.txt

2. Once this is installed, simply run the below command to run and save the model in the directory models/trained:

python3 ImageRecognition

To add more test images for prediction, you can add it in the below location:

test_data/<model_name>

References

• Cifar10 Dataset
• Using TensorFlow & Keras for IR