download网址:https://sourceforge.net/projects/itk-snap/
网页界面:
显示3d的操作方式
1)保存原始图像、预测图像、mask的nii文件
def colorize(n, images, logits, targets):
outputs = torch.argmax(logits, dim=1, keepdim=False).type(torch.uint8)
targets = targets.type(torch.uint8)
outputs = np.array(outputs[0,...]) # choose one to show
targets = np.array(targets[0,...])
img = sitk.GetImageFromArray(outputs * 64)
sitk.WriteImage(img, root + 'img' + str(n) + '_src.nii.gz')
out = sitk