我不熟悉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文件,而不是作者提供的。
根据经验,元流的第一种方法通常效果很好。