Image Recognition

Setup

  1. First install all the required dependencies using the command below:
pip install -r requirements.txt
  1. 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