Skip to content

Instantly share code, notes, and snippets.

View huckl3b3rry87's full-sized avatar

Huckleberry Febbo huckl3b3rry87

  • Honda
  • San Jose
View GitHub Profile
@huckl3b3rry87
huckl3b3rry87 / vgg.py
Created April 20, 2020 21:25
This script will demonstrate how to use a pretrained model, in PyTorch, to make predictions.
"""
This script will demonstrate how to use a pretrained model, in PyTorch,
to make predictions. Specifically, we will be using VGG16 with a cat
image.
References used to make this script:
PyTorch pretrained models doc:
http://pytorch.org/docs/master/torchvision/models.html
PyTorch image transforms example:
http://pytorch.org/tutorials/beginner/data_loading_tutorial.html#transforms