/pytorch-vgg

Primary LanguagePythonMIT LicenseMIT

pytorch-vgg

Some scripts to convert the VGG-16 and VGG-19 models [1] from Caffe to PyTorch.

The converted models can be used with the PyTorch model zoo and are available here:

VGG-16: https://s3-us-west-2.amazonaws.com/jcjohns-models/vgg16-00b39a1b.pth

VGG-19: https://s3-us-west-2.amazonaws.com/jcjohns-models/vgg19-d01eb7cb.pth

These models expect different preprocessing than the other models in the PyTorch model zoo. Images should be in BGR format in the range [0, 255], and the following BGR values should then be subtracted from each pixel: [103.939, 116.779, 123.68]

[1] Karen Simonyan and Andrew Zisserman, "Very Deep Convolutional Networks for Large-Scale Image Recognition", ICLR 2015