分割输出结果，合并原图和标签图片

最新推荐文章于 2024-05-23 08:56:36 发布

无锡太湖学院

最新推荐文章于 2024-05-23 08:56:36 发布

阅读量393

点赞数 10

文章标签：计算机视觉 python numpy 深度学习

本文链接：https://blog.csdn.net/zrz1973854232/article/details/135367091

版权

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)