Skip to content

Instantly share code, notes, and snippets.

@pranjal-joshi
Created April 24, 2017 10:10
Show Gist options
  • Save pranjal-joshi/23b3dd51566b0b95225e12f640ce4d18 to your computer and use it in GitHub Desktop.
Save pranjal-joshi/23b3dd51566b0b95225e12f640ce4d18 to your computer and use it in GitHub Desktop.
b_img = K.variable(preprocess_img(base_path))
r_img = K.variable(preprocess_img(ref_path))
final_img = K.placeholder((1,3,img_nrows,img_ncols))
input_tensor = K.concatenate([b_img, r_img, final_img],axis=0)
model = vgg16.VGG16(input_tensor=input_tensor,weights='imagenet',include_top=False)
output_dict = dict([(layer.name,layer.output) for layer in model.layers])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment