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

【问题标题】：Tensorflow, change checkpoint model(.meta .index .data) to frozen model(.pd)Tensorflow，将检查点模型（.meta .index .data）更改为冻结模型（.pd）
【发布时间】：2019-06-18 20:55:20
【问题描述】：

我不熟悉张量流。

我想将这个网络 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”文件。

我已经测试了以下人给出的想法：

问题是两个给出的想法都失败了，因为 get_checkpoint_state() 和 latest_checkpoint() 给出了 None 值。

这是因为缺少检查点文件造成的吗？

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

任何想法都值得赞赏。

谢谢。

【问题讨论】：

标签： python tensorflow tensorrt

【解决方案1】：

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

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

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

【讨论】：

感谢您的帮助。一个检查点是必要的。这个文件给出了 .meta .index .data 的模型名称。所以我只是自己添加一个新的检查点文件。而且效果很好。
不客气，如果它解决了您的问题，请考虑接受答案