我运行了代码
export_path=os.getcwd()+'\\model\\'+'2016'
with tf.Session(graph=tf.Graph()) as sess:
tf.saved_model.loader.load(sess, ["myTag"], export_path)
graph = tf.get_default_graph()
# print(graph.get_operations())
input = graph.get_tensor_by_name('input:0')
output = graph.get_tensor_by_name('output:0')
# print(sess.run(output,
# feed_dict={input: [test_data[1]]}))
tf.train.write_graph(freeze_session(sess), export_path, "my_model.pb", as_text=False)
并发生以下错误
OSError Traceback (most recent call last)
<ipython-input-44-b154e11ca364> in <module>()
3
4 with tf.Session(graph=tf.Graph()) as sess:
----> 5 tf.saved_model.loader.load(sess, ["myTag"], export_path)
6 graph = tf.get_default_graph()
7 # print(graph.get_operations())
3 frames
/usr/local/lib/python3.6/dist-packages/tensorflow/python/saved_model/loader_impl.py in parse_saved_model(export_dir)
81 (export_dir,
82 constants.SAVED_MODEL_FILENAME_PBTXT,
---> 83 constants.SAVED_MODEL_FILENAME_PB))
84
85
OSError: SavedModel file does not exist at: /content\model\2016/{saved_model.pbtxt|saved_model.pb}
它以前在Windows系统中运行,现在在IOS中运行。我不确定是否是因为这个。任何帮助表示赞赏。谢谢。
答案 0 :(得分:1)
您应避免使用固定字符作为目录分隔符(例如您的示例中的\
),因为该字符在操作系统之间有所不同(请参见os.sep)。
您可能要使用os.path.join或Path来构建路径,从而确保为执行的操作系统创建合适的路径,例如:
export_path = os.path.join(os.getcwd(), 'model', '2016')