image classifier using pytorch
- python 3.7
- numpy
Command line
python train.py dir --gpu --epochs
python predict.py image_path model_path --gpu
A directory containing validation and training data must be available:
dir/train/
Each category within the train folder should be a separate folder containing training images