0

I am not familiar with tensorflow.

I want to transform this network, https://github.com/jiangsutx/SRN-Deblur, from tensorflow to nvidia tensorRT. It need a '.pb' model file, but the project only giving three model files as follow:

  • deblur.model-52300.data-00000-of-00001
  • deblur.model-52300.index
  • deblur.model-52300.meta

So I want to transform these files to a '.pb' file.

I have tested the ideas given by:

The problem is that both gave ideas are failed because of the get_checkpoint_state() and latest_checkpoint() giving None value.

Is this caused by the missing of checkpoint file?

Are there other ways to implement this?

Any idea is appreciated.

Thanks.

Z.LC
  • 1
  • 1

1 Answers1

0

As you can see from their own repo: they use get_checkpoint_state to test a pre-trained model. https://github.com/jiangsutx/SRN-Deblur/blob/master/models/model.py#L245

So I'd say yes it is because of the missing .ckpt file, not provided by the author.

From experience, usually the first method from metaflow works quite well.

Tezirg
  • 1,629
  • 1
  • 10
  • 20
  • Thanks for your help. A chekpoint is neccessary. And this file gives the model name for .meta .index .data. So I just add a new checkpoint file by myself. And it works well. – Z.LC Jan 29 '19 at 01:56
  • You're welcome, please consider accepting the answer if it solved your problem – Tezirg Jan 29 '19 at 11:23