我不熟悉张量流。
我想将这个网络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:如何将 .meta、.data 和 .index 模型文件转换为一个 graph.pb 文件
问题是两个给出的想法都失败了,因为get_checkpoint_state()和latest_checkpoint()给出了 None 值。
这是由于缺少检查点文件引起的吗?
还有其他方法可以实现吗?
任何想法都值得赞赏。
谢谢。