Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
# Load TensorFlow frozen model
TF_FROZEN_MODEL = "mobilenet_v2_1.0_224_frozen.pb"
with open(TF_FROZEN_MODEL, 'rb') as f:
serialized_model = f.read()
tf.reset_default_graph()
graph_definition = tf.GraphDef()
graph_definition.ParseFromString(serialized_model)
with tf.Graph().as_default() as g:
tf.import_graph_def(graph_definition, name='')
# Learn about the model
!cat mobilenet_v2_1.0_224_info.txt
# >> Model: mobilenet_v2_1.0_224
# >> Input: input
# >> Output: MobilenetV2/Predictions/Reshape_1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.