Solved by KFrank in post #10 
              
              
                Hello fyy! 

At this point, the best I can offer you is some general advice. 
First:  Independent of u-net or pytorch or machine learning, you 
need to understand the problem you are trying to solve.  You 
should look at the data you are working with.  Print out some 
images and try to segment them …