Yes, I am working on semantic segmentation on the RGB image.
Since Batch_size = 10 and class_num = 6, my target size is [10,6,256,256].
Yes, I use Resnet architecture, so the output size is [10,2048,256,256].