一杯清酒邀明月
天下本无事,庸人扰之而烦耳。

1、resampleSize
这个重采样不是重采样Spacing ,却是重采样size,为了达到所有的数据一致的size

例如:(880, 880, 12) 采样统一的(880,880,16)

因为假如3DUnet网络数据设置了输入格式为(N, C, 16, xxx,xxx), 可以把每个数据的size都重采样(16,xxx,xxx),然后输入网络。

如果直接设置Size 为(16,xxx,xxx),多余的层SimpleITK会用0来填充,达不到真正的采样。所以通过上面对Spacing采样进行反操作

 1 """
 2 统一Size
 3 X轴和Y轴的Size和Spacing没有变化,
 4 Z轴的Size和Spacing有变化
 5 """
 6 def resampleSize(sitkImage, depth):
 7     #重采样函数
 8     euler3d = sitk.Euler3DTransform()
 9 
10     xsize, ysize, zsize = sitkImage.GetSize()
11     xspacing, yspacing, zspacing = sitkImage.GetSpacing()
12     new_spacing_z = zspacing/(depth/float(zsize))
13 
14     origin = sitkImage.GetOrigin()
15     direction = sitkImage.GetDirection()
16     #根据新的spacing 计算新的size
17     newsize = (xsize,ysize,int(zsize*zspacing/new_spacing_z))
18     newspace = (xspacing, yspacing, new_spacing_z)
19     sitkImage = sitk.Resample(sitkImage,newsize,euler3d,sitk.sitkNearestNeighbor,origin,newspace,direction)
20     return sitkImage
21 
22 DEPTH = 16  #需要重采样Size的层数
23 
24 #读取nifit原数据 ,size为:(880, 880, 12)
25 NifitmPath = 'work/data/nifit/Case14.nii.gz'
26 sitkImage = sitk.ReadImage(NifitmPath)
27 print("重采样前的信息") 
28 print("尺寸:{}".format(sitkImage.GetSize()))
29 print("体素大小(x,y,z):{}".format(sitkImage.GetSpacing()) )
30 
31 print('='*30+'我是分割线'+'='*30)
32 
33 
34 newsitkImage = resampleSize(sitkImage, depth=DEPTH)
35 print("重采样后的信息")
36 print("尺寸:{}".format(newsitkImage.GetSize()))
37 print("体素大小(x,y,z):{}".format(newsitkImage.GetSpacing()) )
1 重采样前的信息
2 尺寸:(880, 880, 12)
3 体素大小(x,y,z):(0.3448275923728943, 0.3448275923728943, 5.499998092651367)
4 ==============================我是分割线==============================
5 重采样后的信息
6 尺寸:(880, 880, 16)
7 体素大小(x,y,z):(0.3448275923728943, 0.3448275923728943, 4.124998569488525)

2、CT图像预处理之重采样

按照指定spacing 重采样数据的缺点就是一般无法重采样到指定尺寸。实际应用中经常遇到需要指定尺寸图像的情况。

 1 def Resampling(img,lable = False):
 2     original_size = img.GetSize() #获取图像原始尺寸
 3     original_spacing = img.GetSpacing() #获取图像原始分辨率
 4     new_spacing = [1, 1, 1] #设置图像新的分辨率为1*1*1
 5 
 6     new_size = [int(round(original_size[0] * (original_spacing[0] /1))),
 7                 int(round(original_size[1] * (original_spacing[1] / 1))),
 8                 int(round(original_size[2] * (original_spacing[2] / 1)))] #计算图像在新的分辨率下尺寸大小
 9     resampleSliceFilter = sitk.ResampleImageFilter() #初始化
10     if lable == False:
11         Resampleimage = resampleSliceFilter.Execute(img, new_size, sitk.Transform(), sitk.sitkBSpline,
12                                                 img.GetOrigin(), new_spacing, img.GetDirection(), 0,
13                                                 img.GetPixelIDValue())
14         ResampleimageArray = sitk.GetArrayFromImage(Resampleimage)
15         ResampleimageArray[ResampleimageArray < 0] = 0 #将图中小于0的元素置为0
16     else:# for label, should use sitk.sitkLinear to make sure the original and resampled label are the same!!!
17         Resampleimage = resampleSliceFilter.Execute(img, new_size, sitk.Transform(), sitk.sitkLinear,
18                                                     img.GetOrigin(), new_spacing, img.GetDirection(), 0,
19                                                     img.GetPixelIDValue())
20         ResampleimageArray = sitk.GetArrayFromImage(Resampleimage)

3、SKIMAGE.TRANSFORM-两种缩放皆可

 1 # 1、按目标大小缩放resize
 2 skimage.transform.resize(image, output_shape, order=1, mode='reflect', cval=0, clip=True, preserve_range=False, anti_aliasing=True, anti_aliasing_sigma=None)
 3 # 2、按缩放因子缩放rescale
 4 skimage.transform.rescale(image, scale, order=1, mode='reflect', cval=0, clip=True, preserve_range=False, multichannel=False, anti_aliasing=True, anti_aliasing_sigma=None)
 5 
 6 # The order of interpolation. The order has to be in the range 0-5:
 7 # 0: Nearest-neighbor
 8 # 1: Bi-linear (default)
 9 # 2: Bi-quadratic
10 # 3: Bi-cubic
11 # 4: Bi-quartic
12 # 5: Bi-quintic

4、Python SimpleItk库的医学图像重采样(resample)实现代码
  需求:已有配准好的CT以及PET图像,而金标准label是在CT上勾画的,因此有一些简单的需求,一种是把PET图像重采样到与CT图像一样的大小(比如从192×192×371到512×512×484),或者把金标准Mask降到同PET的大小(即反过来)。

  怎么找的方法:ITK-SNAP(3.8版本)的读图功能是支持不同大小、spacing、origin、direction的图像一起展示的,软件会相当于resample后续加入的图像; 因此,已知ITK存在解决的方法了,剩下就是找对应代码。

 1 import SimpleITK as sitk
 2 
 3 def resize_image_itk(ori_img, target_img, resamplemethod=sitk.sitkNearestNeighbor):
 4     """
 5     用itk方法将原始图像resample到与目标图像一致
 6     :param ori_img: 原始需要对齐的itk图像
 7     :param target_img: 要对齐的目标itk图像
 8     :param resamplemethod: itk插值方法: sitk.sitkLinear-线性  sitk.sitkNearestNeighbor-最近邻
 9     :return:img_res_itk: 重采样好的itk图像
10     使用示范:
11     import SimpleITK as sitk
12     target_img = sitk.ReadImage(target_img_file)
13     ori_img = sitk.ReadImage(ori_img_file)
14     img_r = resize_image_itk(ori_img, target_img, resamplemethod=sitk.sitkLinear)
15     """
16     target_Size = target_img.GetSize()      # 目标图像大小  [x,y,z]
17     target_Spacing = target_img.GetSpacing()   # 目标的体素块尺寸    [x,y,z]
18     target_origin = target_img.GetOrigin()      # 目标的起点 [x,y,z]
19     target_direction = target_img.GetDirection()  # 目标的方向 [冠,矢,横]=[z,y,x]
20 
21     # itk的方法进行resample
22     resampler = sitk.ResampleImageFilter()
23     resampler.SetReferenceImage(ori_img)  # 需要重新采样的目标图像
24     # 设置目标图像的信息
25     resampler.SetSize(target_Size)        # 目标图像大小
26     resampler.SetOutputOrigin(target_origin)
27     resampler.SetOutputDirection(target_direction)
28     resampler.SetOutputSpacing(target_Spacing)
29     # 根据需要重采样图像的情况设置不同的dype
30     if resamplemethod == sitk.sitkNearestNeighbor:
31         resampler.SetOutputPixelType(sitk.sitkUInt8)   # 近邻插值用于mask的,保存uint8
32     else:
33         resampler.SetOutputPixelType(sitk.sitkFloat32)  # 线性插值用于PET/CT/MRI之类的,保存float32
34     resampler.SetTransform(sitk.Transform(3, sitk.sitkIdentity))    
35     resampler.SetInterpolator(resamplemethod)
36     itk_img_resampled = resampler.Execute(ori_img)  # 得到重新采样后的图像
37     return itk_img_resampled

5、使用SimpleITK对volume文件进行插值处理

  在进行dicom文件分析时,往往会对其进行插值或者重采样处理,可以使用SimpleITK轻松完成。

 1 import SimpleITK as sitk
 2 
 3 def getDicomSeriesVolumeImage(folderPath):
 4     '''
 5     读取dicom序列文件
 6     :para folderPath:dicom文件夹
 7     :return volumeImage:volume
 8     '''
 9     reader = sitk.ImageSeriesReader()
10     dicomNames = reader.GetGDCMSeriesFileNames(folderPath)
11     reader.SetFileNames(dicomNames)
12     volumeImage = reader.Execute()
13 
14     return volumeImage
15 
16 def interpolate(volumeImage, newSpacing):
17     '''
18     插值处理
19     :para volumeImage:volume格式
20     :para newSpaceing:想要插值或者重采样处理后得到的三个物理间隔,格式为列表
21     :return newVolumeImage:插值处理后的到的volume文件
22     '''
23     resampleFilter = sitk.ResampleImageFilter()
24     resampleFilter.SetInterpolator(sitk.sitkLinear)     ##此处为线性插值,其他插值方式可以去官网查询
25     resampleFilter.SetOutputDirection(volumeImage.GetDirection())
26     resampleFilter.SetOutputOrigin(volumeImage.GetOrigin())
27 
28     newSpacing = np.array(newSpacing, float)
29     newSize = volumeImage.GetSize() / newSpacing * volumeImage.GetSpacing()
30     newSize = newSize.astype(np.int)
31 
32     resampleFilter.SetSize(newSize.tolist())
33     resampleFilter.SetOutputSpacing(newSpacing)
34     newVolumeImage = resampleFilter.Execute(volumeImage)
35 
36     return newVolumeImage

6、医学影像重采样
1. 重采样
  在医学图像预处理的过程中,常常需要对数据进行重采样,采样到我们需要的尺度。因为医学图像是和真实空间的物理尺寸相对应的。例如一个体素的大小是0.97mm 0.97mm 2.5mm,我们想将体素块的大小采样到1mm 1mm 1mm,这样可以帮助我们进行下一步处理。利用 sitk 可以完成这一项工作。

2. 代码

 1 import SimpleITK as sitk
 2 """
 3 resample
 4 """
 5 
 6 def resampleVolume(outspacing,vol):
 7     """
 8     将体数据重采样的指定的spacing大小\n
 9     paras:
10     outpacing:指定的spacing,例如[1,1,1]
11     vol:sitk读取的image信息,这里是体数据\n
12     return:重采样后的数据
13     """
14     outsize = [0,0,0]
15     inputspacing = 0
16     inputsize = 0
17     inputorigin = [0,0,0]
18     inputdir = [0,0,0]
19 
20     #读取文件的size和spacing信息
21     
22     inputsize = vol.GetSize()
23     inputspacing = vol.GetSpacing()
24 
25     transform = sitk.Transform()
26     transform.SetIdentity()
27     #计算改变spacing后的size,用物理尺寸/体素的大小
28     outsize[0] = int(inputsize[0]*inputspacing[0]/outspacing[0] + 0.5)
29     outsize[1] = int(inputsize[1]*inputspacing[1]/outspacing[1] + 0.5)
30     outsize[2] = int(inputsize[2]*inputspacing[2]/outspacing[2] + 0.5)
31 
32     #设定重采样的一些参数
33     resampler = sitk.ResampleImageFilter()
34     resampler.SetTransform(transform)
35     resampler.SetInterpolator(sitk.sitkLinear)
36     resampler.SetOutputOrigin(vol.GetOrigin())
37     resampler.SetOutputSpacing(outspacing)
38     resampler.SetOutputDirection(vol.GetDirection())
39     resampler.SetSize(outsize)
40     newvol = resampler.Execute(vol)
41     return newvol    
42 
43 def main():
44     #读文件
45     vol = sitk.Image(sitk.ReadImage("input.mha"))
46 
47     #重采样
48     newvol = resampleVolume([1,1,1],vol)
49 
50     #写文件
51     wriiter = sitk.ImageFileWriter()
52     wriiter.SetFileName("output.mha")
53     wriiter.Execute(newvol)

 

posted on 2023-07-13 14:42  一杯清酒邀明月  阅读(759)  评论(0编辑  收藏  举报