python+simpleITK(sitk)用来resize图像大小

import SimpleITK as sitk
import os
from glob import glob


def resample_sitkimage(sitk_image_,new_size_):
    org_spacing = sitk_image_.GetSpacing()
    org_size = sitk_image_.GetSize()
    print(f'before resize, spacing: {
      
      org_spacing}, size: {
      
      org_size}')
    new_space = [(old_size*old_space/new_size) for old_size,old_space,new_size in zip(org_size,org_spacing,new_size_)]

    resampled_img=sitk.Resample(
        sitk_image_,
        new_size_,
        sitk.Transform(),
        sitk.sitkLinear,
        sitk_image_.GetOrigin(),
        new_space,
        sitk_image_.GetDirection(),
        0,
        sitk_image_.GetPixelID()
        )
    return resampled_img


if __name__ == '__main__':
    work_dir = "/home/data/hablee_data_dir/aorta_data/"
    nii_paths = glob(work_dir+"*.nii.gz")
   
    for path in nii_paths:
        image = sitk.ReadImage(path)
        desired_size = [size//2 for size in image.GetSize()]
        
        new_image = resample_sitkimage(image,desired_size)
        new_spacing = new_image.GetSpacing()
        new_size = new_image.GetSize()
        print(f'after resize, spacing: {
      
      new_spacing}, size: {
      
      new_size}')

猜你喜欢

转载自blog.csdn.net/sdhdsf132452/article/details/129680816