Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@khanhnamle1994

khanhnamle1994/vgg16.py

Last active May 8, 2018 13:47
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save khanhnamle1994/9a99055d79a2fbe093ca68e74078bb31 to your computer and use it in GitHub Desktop.
Save khanhnamle1994/9a99055d79a2fbe093ca68e74078bb31 to your computer and use it in GitHub Desktop.
Download ZIP
FCN - Load Pretrained VGG Model into TensorFlow.
Raw
vgg16.py
def load_vgg(sess, vgg_path):
# load the model and weights
model = tf.saved_model.loader.load(sess, ['vgg16'], vgg_path)
# Get Tensors to be returned from graph
graph = tf.get_default_graph()
image_input = graph.get_tensor_by_name('image_input:0')
keep_prob = graph.get_tensor_by_name('keep_prob:0')
layer3 = graph.get_tensor_by_name('layer3_out:0')
layer4 = graph.get_tensor_by_name('layer4_out:0')
layer7 = graph.get_tensor_by_name('layer7_out:0')
return image_input, keep_prob, layer3, layer4, layer7
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment