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.