Image classifier based on pytorch.
- https://pytorch.org/ (use with conda)
Expected folder structure of the images:
- image-folder/
- test/
- cat1/
- cat2/
- cat3/
- valid/
- cat1/
- cat2/
- cat3/
- test/
{"cat1": "Foo Bar", "cat2": "John Doe"}python train.py ./image-folder
python train.py ./path-to-image/image.jpg ./path-to-trainer-checkpoint/model.pth ./path-to-category-json/cat.json