Hi, I managed to make the system learn in the gist, so the problem is related to the network architecture. I think that the problem is in the classification layer, which applies a 1x1 convolution from 512 filters to num_classes. The output of all the the network is always equal between all the pixels, so something wrong is going there that i cannot see.
Any help would be appreciated
thanks