Image Classification for any category using pretrained networks from pytorch (torchvision)
- train network on a given dataset including train, validation and test images (see train.py -h for instructions)
- check your trained model (see check.py -h for instructions)
- predict class membership of a single image, given the trained network (see predict.py -h for instructions)