TF2.0 创建单个 pb 文件而不需要其他文件夹或文件？

Question

我搜索并阅读了几篇文章，但我找不到任何提示来创建一个没有变量的单个 pb 文件，assets 文件夹。

我的tensorflow版本是2.0.0

https://www.tensorflow.org/guide/saved_model

从这篇文档中，我可以从检查点创建一个 pb 文件。 我在我的文件夹中找到了一个具有这种结构的 pb 文件。

• pb_file_parent_dir/assets

• pb_file_parent_dir/saved_model.pb

• pb_file_parent_dir/variables/

我想创建一个 pb 文件（saved_model.pb），但我有 2 个额外的文件夹（资产和变量）。

如果有办法创建单个 pb 文件，请帮帮我。我想为部署目的创建一个 pb 文件。

已编辑

我尝试使用 'tf.keras.models.save_model' 和 https://www.tensorflow.org/api_docs/python/tf/keras/Model#save。我得到了与上面相同的结果，而不是单个 pb 文件。

谢谢，祝你有个愉快的一天。

Answer 1

TensorFlow 2.0 不再使用 'frozen' 单个文件 .pb 来存储您的模型，而是使用 SavedModel 格式 https://www.tensorflow.org/guide/saved_model (which is what you have already with a .pb and variables/assets folders) or alternatively you can use Keras .h5 format https://www.tensorflow.org/tutorials/keras/save_and_load#hdf5_format。

您仍然可以使用 compat 模块在 tf2.0 中创建冻结的 .pb 文件，https://www.tensorflow.org/api_docs/python/tf/compat/v1/graph_util/convert_variables_to_constants但是您必须使用所有兼容的东西，比如会话（这不是 TensorFlow 2.0 的处理方式并且与 keras 不兼容 api 我相信）。