• Stars
    star
    218
  • Rank 175,629 (Top 4 %)
  • Language
    Python
  • License
    MIT License
  • Created over 7 years ago
  • Updated about 4 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

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://web.eecs.umich.edu/~justincj/models/vgg16-00b39a1b.pth

VGG-19: https://web.eecs.umich.edu/~justincj/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