Hello,
I trained the U-Net for several images from a datasef which were converted into pytorch-tensors in an ID_list.txt.
The model works fine and even the evaluation on the other images looks good.
My question now is how I can apply the state-dict of the model that I trained and saved to one single (external) 256x256 image ? How does this single 256x256 image need to be converted to go through the U-net and be able to use with the saved state_dict?