In [2]:
import tensorflow as tf
import numpy as np
from tensorflow.python.platform import gfile
model, labels = ('./inception5h/tensorflow_inception_graph.pb',
'./inception5h/imagenet_comp_graph_label_strings.txt')
with gfile.GFile(model, 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
tf.import_graph_def(graph_def=graph_def,name='inception')