tf_unet
tf_unet copied to clipboard
Questions of a noob
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.
tf_unet stores the intermediate model checkpoints under the output_path. You should be able to reload the tf graph from there
I have my computer busy right now but i will be something like:
with tf.Session() as sess: unet.restore(sess,path)
(?)
Sounds about right
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")
Hmm thats something I haven't seen before