tf_unet
tf_unet copied to clipboard
How to get output of a batch of input files
Hello,
I am simplifying my question
I have images in a folder that I want to use my trained model on to segment. I am really confused on how to do this and have it output images. I would greatly appreciate any help.
You can load your data using the same data provider mechanism - so you ensure that the same pre-processing steps are being applied
ok, well once I load them and then use net.predict(), how can I then safely convert the batch of tensors to rob images and save them to image files? I tried doing this with my own code and got all black or all white images.