本演示使用Google Inception模型对在命令行中传递的图像文件进行分类。
函数作用:载入图片,并返回图片。
函数声明情况如下:
函数声明文件如下:
函数流程图如下:
函数原始代码如下:
def load_graph(model_file):
graph = tf.Graph()
graph_def = tf.GraphDef()
with open(model_file, "rb") as f:
graph_def.ParseFromString(f.read())
with graph.as_default():
tf.import_graph_def(graph_def)
return graph