tf_unet icon indicating copy to clipboard operation
tf_unet copied to clipboard

Questions of a noob

Open LauraLaureus opened this issue 8 years ago • 5 comments

Hello everyone. I'm quite noob at tensorflow and unet. I have been training unet for three days and My computer is not the best computation beast you can find in the market so I would like to have the chance to retrieve the results of the unet to use it in other code.

How can i do this?

Thank you for your time everyone.

LauraLaureus avatar Sep 08 '17 09:09 LauraLaureus

tf_unet stores the intermediate model checkpoints under the output_path. You should be able to reload the tf graph from there

jakeret avatar Sep 08 '17 14:09 jakeret

I have my computer busy right now but i will be something like:

with tf.Session() as sess: unet.restore(sess,path)

(?)

LauraLaureus avatar Sep 08 '17 15:09 LauraLaureus

Sounds about right

jakeret avatar Sep 08 '17 15:09 jakeret

I have problems trying to resore. Somewhere in tensorflow the permission changes and cannot open the file. Any ideas @jakeret ?

This is my code: net = unet.Unet(channels=channels, n_class=n_class, layers=3, features_root=16, cost="dice_coefficient")

sess = tf.Session()

net.restore(sess,model_path = "./unet_trained")

LauraLaureus avatar Sep 09 '17 10:09 LauraLaureus

Hmm thats something I haven't seen before

jakeret avatar Sep 10 '17 12:09 jakeret