download网址:https://sourceforge.net/projects/itk-snap/
https://sourceforge.net/projects/itk-snap/files/itk-snap/3.8.0/itksnap-3.8.0-20190612-win64.exe/download
网页界面:
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.GetImageFromArray(outputs * 64) sitk.WriteImage(out, root+'img'+str(n)+'_pred.nii.gz') mask = sitk.GetImageFromArray(targets * 64) sitk.WriteImage(mask, root + 'img' + str(n) + '_mask.nii.gz') print('save done') return 02)在软件中打开img_src.nii,作为主图像
3)将pred或者mask图像拖到软件中,作为segmentation导入
4)点击下方的continue update