File "/disk/sdc/shejiao/Pytorch-UNet-master/evaluate.py", line 19, in evaluate mask_true = F.one_hot(mask_true, net.n_classes).permute(0, 3, 1, 2).float() RuntimeError: number of dims don't match in permute

[DonMuv]

File "/disk/sdc/shejiao/Pytorch-UNet-master/evaluate.py", line 19, in evaluate mask_true = F.one_hot(mask_true, net.n_classes).permute(0, 3, 1, 2).float() RuntimeError: number of dims don't match in permute

[jiangzhen3s]

mask = np.zeros((W,H,3),dtype=np.uint8) ...... maskImg = Image.fromarray(mask) maskImg.save(dir+'/train/mask/'+imgPath[0:-4]+'_mask.gif')

it works for me.