fcn = vgg16(pretrained=True, fully_conv=True)
fcn = fcn.eval()

from torch.autograd import Variable
import torch.nn as nn

import skimage.io as io
import skimage.transform
import numpy as np
import torch

img = skimage.img_as_float(io.imread('bus.jpg'))
img = img - [0.485, 0.456, 0.406]
img = img / [0.229, 0.224, 0.225]
img = img.transpose((2, 0, 1))
img = np.expand_dims(img, 0)
img = img.astype(np.float32)
final_img = torch.from_numpy(img)

fcn.cuda()
final_img = Variable(final_img.cuda())
res = fcn(final_img)

boo = res.cpu().data.numpy()
final = boo.squeeze().argmax(axis=0)
# 779 -- is a class for a bus from imagenet
io.imshow(final == 779)