0

我不熟悉张量流。

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

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

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

任何想法都值得赞赏。

谢谢。

4

1 回答 1

0

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

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

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

于 2019-01-25T10:37:35.110 回答