Skip to content

Instantly share code, notes, and snippets.

@richardgrantserverless
Created May 31, 2022 16:40
Show Gist options
  • Save richardgrantserverless/ec617f8764f8a36042b80beef59eb04f to your computer and use it in GitHub Desktop.
Save richardgrantserverless/ec617f8764f8a36042b80beef59eb04f to your computer and use it in GitHub Desktop.
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