Hello,
I am following your link. I have few queries onto this:


I have 3 classes, even though I tried to change the number of out_channels to 3, still model.eval() shows final layer with 21 classes. Please clarify this.


It has batch size 4. I am trying to train it on single image. So when I change batch size to 1, it is throwing an error.


Can you help to decode the result into a mask? I dont know how to do that.


Please help in this regard. Thank you.