Skip to content

Instantly share code, notes, and snippets.

@cancan101
Created November 23, 2016 00:09
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save cancan101/31df34ca5dd971338cad8ca85bc1d8e2 to your computer and use it in GitHub Desktop.
Save cancan101/31df34ca5dd971338cad8ca85bc1d8e2 to your computer and use it in GitHub Desktop.
sess, meta_graph_def = session_bundle.load_session_bundle_from_path(input)
signatures_any = sess.graph.get_collection(constants.SIGNATURES_KEY)
signatures = manifest_pb2.Signatures()
signatures_any[0].Unpack(signatures)
inputs = signatures.named_signatures['inputs'].generic_signature
input_name = inputs.map['images'].tensor_name
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment