This repo is part of the project submission for the Udacity nano degree program "AI Programming with Python". The purpose of the project is to first develop code for an image classifier built with PyTorch, then convert it into a command line application.
- Clone the repository to your local machine
- Create a virtual environment and activate it
- Install the dependencies using the command
pip install -r requirements.txt
- Work on Jupyter notebook "Image Classifier Project.ipynb"
- Or use
train.py
andpredict.py
to run the command line application. Use with -h argument to see the available arguments