import cv2 from PIL import Image import numpy as np import SimpleITK as sitk import imageio import nibabel as nib # def changeLabel(mask_image , ): img_path_1 = "Case1.nii.gz" mask_path_1 = "mask_case1.nii.gz" # mask_path_1 = "resize_mask.nii.gz" img_1 = sitk.GetArrayFromImage(sitk.ReadImage(img_path_1)) mask_2 = sitk.GetArrayFromImage(sitk.ReadImage(mask_path_1)) original_image = img_1[7, :, :] # original_image = np.rot90(original_image) # original_image = np.rot90(original_image) original_image = np.flip(original_image , axis = 0) # original_image = np.expand_dims(original_image , axis = 2) mask_image = mask_2[7, :, :] mask_image = np.flip(mask_image , axis = 0) mask_image =np.where(mask_image > 2, 0, mask_image) mask_image =np.where(mask_image < 2, 0, mask_image) mask_image =np.where(mask_image == 2, 255, mask_image) #更改标签 imageio.imwrite("imageio.png", original_image) imageio.imwrite("mask.png", mask_image) original_image = cv2.imread("imageio.png", cv2.IMREAD_GRAYSCALE) mask_image = cv2.imread("mask.png", cv2.IMREAD_GRAYSCALE) # 确保图像大小相同 original_image = cv2.resize(original_image, (mask_image.shape[1], mask_image.shape[0])) # 将白色背景转换为蓝色 result_image = cv2.cvtColor(original_image, cv2.COLOR_GRAY2BGR) result_image[mask_image == 255] = [255, 255, 100] # 将黑色背景设置为透明 result_image[np.logical_and(mask_image != 255, mask_image != 0)] = [0, 0, 0] save_path = "out.png" cv2.imwrite(save_path, result_image)
分割输出结果,合并原图和标签图片
最新推荐文章于 2024-10-01 13:07:52 发布