Hello,
The goal of my model is at the end to classify each pixel to the correct (same position) pixel in the label. I have output images of shape [1, 3, 192, 144] for my model, and my labels the shape [1, 192, 144]. I can’t seem to understand how i am going to compute the loss and train my model. Could someone explain it a bit more clearly or point me to some documentation ? (I already searched but no one seems to be doing a training and loss computation compatible).
For now I am doing a crossentropy loss computation but I don’t think it is the correct method.