Hello
i have the same problem, if you can help me ?
i’m working on the IRM’s data, i had normalize my inputs data at the beginning, and I didn’t normalize the groudtruth segmentation !!!
the problem is that I have 4 class [0: bachground, 1: LV, 2: MYO, 3: RV]
but in output after training my model I only have the backgroud
i’m ploting that like this :
for i in range(0,len(outputs)):
_imshow(inputs[i][0].cpu().numpy(),labels[i].cpu().numpy(),torch.argmax(outputs[i],0).cpu().numpy())
and i get that
Capture955×373 52.9 KB
thanks in advance, hope you could help me