SimpleITK图像增强

以下是使用SimpleITK进行图像增强的示例代码：

1.对比度增强

import SimpleITK as sitk

# 读取图像
image = sitk.ReadImage("input_image.mha")

# 对比度增强
enhanced_image = sitk.IntensityWindowing(image, windowMinimum=100, windowMaximum=200, outputMinimum=0.0, outputMaximum=1.0)

# 保存增强后的图像
sitk.WriteImage(enhanced_image, "enhanced_image.mha")

2.直方图均衡化

import SimpleITK as sitk
# 读取图像
image = sitk.ReadImage("input_image.mha")

# 直方图均衡化
equalized_image = sitk.EqualizeHist(image)

# 保存均衡化后的图像
sitk.WriteImage(equalized_image, "equalized_image.mha")

3.对数变换

import SimpleITK as sitk
import numpy as np

# 读取图像
image = sitk.ReadImage("input_image.mha")

# 将图像转换为Numpy数组
np_image = sitk.GetArrayFromImage(image)

# 对数变换
np_log_image = np.log1p(np_image - np.min(np_image))
log_image = sitk.GetImageFromArray(np_log_image)
log_image.SetOrigin(image.GetOrigin())
log_image.SetDirection(image.GetDirection())
log_image.SetSpacing(image.GetSpacing())

# 保存对数变换后的图像
sitk.WriteImage(log_image, "log_image.mha")