Tensorflow2.0 Tensorflow 2.0冻结图在tf.keras保存的模型上失败 < >我将一个经过训练的TF.KARAS模型转换成TF冻结模型，用C++ API使用。我在TF2.0上冻结模型时出错 model_path = '/home/Desktop/model.hdf5' model = tf.keras.models.load_model(model_path) tf.keras.experimental.export_savedmodel(model,newdir)

在此之后，将创建一个包含文件[checkpoint，variables.data-00000-of-00001，variables.index]的variables文件夹，保存了_model.pb并在newdir中创建了Assets文件夹

我正在尝试使用保存的\u model.pb和variables.data-00000-of-00001文件来获取单个.pb冻结的\u图

python /home/tensorflow/python/tools/freeze_graph.py --input_graph=/home/Desktop/tf_models/saved_model.pb --input_checkpoint=/home/Desktop/tf_models/variables/variables.data-00000-of-00001 --output_graph=/home/Desktop/tf_models/frozen_graph.pb --output_node_names=classes --input_binary=true

我希望有一个冻结的.pb文件，但却遇到如下错误：

回溯（最近一次调用上次）：文件 “/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site-packages/tensorflow/python/tools/freeze_-graph.py”， 第492行，在run_main（）文件中 “/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site-packages/tensorflow/python/tools/freeze_-graph.py”， 第489行，在run\u main app.run中（main=my\u main，argv=[sys.argv[0]]+ 未解析）文件 “/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site packages/tensorflow/python/platform/app.py”， 第40行，运行中 _运行（main=main，argv=argv，flags\u parser=\u parse\u flags\u undef）文件 “/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site packages/absl/app.py”， 第300行，运行中 _运行_main（main，args）文件“/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site packages/absl/app.py”， 第251行，在_run_main sys.exit（main（argv））文件中 “/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site-packages/tensorflow/python/tools/freeze_-graph.py”， 第488行，my_main=lambda unused_args:main（unused_args，flags） 文件 “/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site-packages/tensorflow/python/tools/freeze_-graph.py”， 第382行，在main flags.saved_model_tags，checkpoint_version）文件中 “/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site-packages/tensorflow/python/tools/freeze_-graph.py”， 第341行，在冻结图输入图定义中= _解析输入图原型（输入图，输入二进制）文件“/home/vsrira10/anaconda2/envs/tf2/lib/python3.6/site packages/tensorflow/python/tools/freeze\u graph.py”， 第252行，在_parse_input_graph_proto中 input_graph_def.ParseFromString（f.read（）） google.protobuf.message.DecodeError:解析消息时出错

---

我愿意接受使用冻结图脚本的其他建议。谢谢。

您知道如何在tf2中冻结已保存的\u模型了吗？我也想知道怎么做