Tensorflow，将检查点模型（.meta .index .data）更改为冻结模型（.pd）

Question

我对张量流不熟悉。

我想将此网络https://github.com/jiangsutx/SRN-Deblur从tensorflow转换为nvidia tensorRT。它需要一个'.pb'模型文件，但是该项目仅提供了三个模型文件，如下所示：

• deblur.model-52300.data-00000-of-00001
• deblur.model-52300.index
• deblur.model-52300.meta

所以我想将这些文件转换为'.pb'文件。

我已经测试了以下观点：

• https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-serve-it-with-a-python-api-d4f3596b3adc
• Tensorflow: How to convert .meta, .data and .index model files into one graph.pb file

问题在于，由于 get_checkpoint_state（）和 latest_checkpoint（）没有给出值，两个给出的想法都失败了。

这是由于缺少检查点文件引起的吗？

还有其他方法可以实现吗？

任何想法都值得赞赏。

谢谢。

Answer 1

从他们自己的仓库中可以看到：他们使用get_checkpoint_state测试预先训练的模型。 https://github.com/jiangsutx/SRN-Deblur/blob/master/models/model.py#L245

所以我说是的，因为缺少作者未提供的.ckpt文件。

根据经验，通常元流中的第一种方法效果很好。