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



我不熟悉tensorflow。

我想改变这个网络,https://github.com/jiangsutx/SRN-Deblur,从tensorflow到nvidia tensorRT。它需要一个".pb"模型文件,但该项目只提供了三个模型文件,如下所示:

  • deblur.model-52300.data-00000-of00001
  • 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:如何将.meta、.data和.index模型文件转换为一个graph.pb文件

问题是,由于get_checkpoint_state()latest_checkpoint()出None值,两个给定的想法都失败了。

这是由于检查点文件丢失造成的吗?

有其他方法可以实现吗?

任何想法都值得赞赏。

谢谢。

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

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

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

最新更新