我对张量流不熟悉。
我想将此网络https://github.com/jiangsutx/SRN-Deblur从tensorflow转换为nvidia tensorRT。它需要一个'.pb'模型文件,但是该项目仅提供了三个模型文件,如下所示:
所以我想将这些文件转换为'.pb'文件。
我已经测试了以下观点:
问题在于,由于 get_checkpoint_state()和 latest_checkpoint()没有给出值,两个给出的想法都失败了。
这是由于缺少检查点文件引起的吗?
还有其他方法可以实现吗?
任何想法都值得赞赏。
谢谢。
答案 0 :(得分:0)
从他们自己的仓库中可以看到:他们使用get_checkpoint_state
测试预先训练的模型。
https://github.com/jiangsutx/SRN-Deblur/blob/master/models/model.py#L245
所以我说是的,因为缺少作者未提供的.ckpt文件。
根据经验,通常元流中的第一种方法效果很好。