Tensorflow,将检查点模型(.meta .index .data)更改为冻结模型(.pd)

时间:2019-01-25 02:33:53

标签: python tensorflow tensorrt

我对张量流不熟悉。

我想将此网络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()没有给出值,两个给出的想法都失败了。

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

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

任何想法都值得赞赏。

谢谢。

1 个答案:

答案 0 :(得分:0)

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

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

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