Download VGG models with their FC Layers stripped, and convert them to a PyTorch compatible format

download_models.sh

cd models

echo "Downloading model conversion scripts..."
wget -c https://raw.github.com/jcjohnson/cnn-benchmarks/master/convert_model.lua
wget -c https://raw.github.com/jcjohnson/cnn-benchmarks/master/utils.lua

echo "Downloading model prototxt files..."
wget -c https://raw.githubusercontent.com/crowsonkb/style_transfer/master/vgg16.prototxt
wget -c https://raw.githubusercontent.com/crowsonkb/style_transfer/master/vgg19.prototxt

echo "Downloading model files..."
wget -c https://style-transfer.s3-us-west-2.amazonaws.com/vgg16.caffemodel
wget -c https://style-transfer.s3-us-west-2.amazonaws.com/vgg19.caffemodel

echo "Converting caffemodels to PyTorch compatible format..."

th convert_model.lua -input_prototxt vgg16.prototxt -input_caffemodel vgg16.caffemodel -output_t7 vgg16.t7
th convert_model.lua -input_prototxt vgg19.prototxt -input_caffemodel vgg19.caffemodel -output_t7 vgg19.t7

echo "Models sucessfully converted!"
cd ..