Python application that can train an image classifier on any dataset , then predict new images using the trained model
- Download the zip file / clone the repo and open it with your preferred coding environment.
- Add the folder with your labelled image dataset that you want to train your model with on the same folder as the other files.
- Go to the command shell and type in: python train.py directory_name_of_your_images
- After the model is trained. Now it is the time to check if it works. In command shell write: python predict.py /path/to/image checkpoint