@lxtGH I used res.argmax(-1) given that res is my prediction of 21 channels. The outputt generated from res.argmax(-1) is 0 and get the output in following way:

pred.png1432×786 40.9 KB

This is my model structure for FCN-32:

model1.png1343×758 93.6 KB
 
model2.png1343×758 81.8 KB

These are my hyper-parameter settings:

hyper.png1469×323 38.2 KB

Have I build the model correctly? why am I getting 0 predictions?