Predictions on original Imagenet categories

Here is a notebook that given a set of validation images and an imagenet model will produce predictions for the original imagenet categories. This may be useful if you are trying to use these predictions as input to another model.

https://github.com/yanneta/pytorch-tutorials/blob/master/predict_from_pretrained.ipynb
4 Likes

Thanks @yinterian, this is great :slight_smile: