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