Skip to content

Instantly share code, notes, and snippets.

@richardgrantserverless
Created May 31, 2022 16:40
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Embed
What would you like to do?
softmax_tensor = sess.graph.get_tensor_by_name('softmax:0')
predictions = sess.run(softmax_tensor,
{'DecodeJpeg/contents:0': image_data})
predictions = np.squeeze(predictions)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment