CODEWITHSUNDEEP
pythontensorflow
Gabriele S
Gabriele S

Reputation: 29

Convert model.ckpt to usable model

I'm training an inception model from scratch using the scripts provided here. The output of the training are these files:

checkpoint
events.out.tfevents.1499334145.fdbf-Dell
model.ckpt-5000.data-00000-of-00001
model.ckpt-5000.index
model.ckpt-5000.meta
...
model.ckpt-25000.data-00000-of-00001
model.ckpt-25000.index
model.ckpt-25000.meta

Does someone have a script to convert these files in something I can use to classify my images? I have already tried to modify the inception_train.py file to output the graph.pb, but nothing happens...

Any help would be appreciated, thank you!

Upvotes: 0

Views: 2777

Answers (1)

gdelab
gdelab

Reputation: 6220

How to use your checkpoint directly is explained here.

To create a .pb file, you'll have to freeze the graph, as explained here.

To create the initial .pb (or .pbtxt) file needed for freeze_graph, you can use tf.train.write_graph()

Upvotes: 1

Related Questions