SimpleITK Dicom序列数据读取

最新推荐文章于 2024-05-15 22:12:36 发布

mikeyNa

最新推荐文章于 2024-05-15 22:12:36 发布

阅读量839

点赞数

分类专栏： ITK学习笔记文章标签： ITK数据读取

本文链接：https://blog.csdn.net/whunamikey/article/details/121764913

版权

ITK学习笔记专栏收录该内容

1 篇文章 0 订阅

订阅专栏

import sys
import os
import numpy as np 
import SimpleITK as sitk

def sitk_dcms_read2(dcm_dir):
    reader = sitk.ImageSeriesReader();
    seriesIDs = reader.GetGDCMSeriesIDs(dcm_dir)
    dcm_series = reader.GetGDCMSeriesFileNames(dcm_dir, seriesIDs[0])
    reader.SetFileNames(dcm_deries)
    itkimage =reader.Execute()
    numpyImage = sitk.GetArrayFromImage(itkimage)
    numpyOrigin = np.array(list(reversed(itkimage.GetOrigin())))
    numpySpacing = np.array(list(reversed(itkimage.GetSpacing())))
    return numpyImage ,numpyOrigin,numpySpacing


def save_itk(image, origin, sapce, filename):
     if type(origin) != tuple:
        if type(origin) == list:
            origin = tuple(reversed(origin))
        else:
            origin = tuple(reversed(origin.tolist()))
    if type(spacing) != tuple:
        if type(spacing) == list:
            spacing = tuple(reversed(spacing))
        else:
            spacing = tuple(reversed(spacing.tolist()))
    itkimage = sitk.GetImageFromArray(image, isVector=False)
    itkimage.SetSpacing(spacing)
    itkimage.SetOrigin(origin)
    sitk.WriteImage(itkimage, filename, True)
    

###示例
raw_path = "D://case01"
filename = "D://case01.vtk"
imgs, origin, spacing = sitk_dcms_red2(filename)
save_itk(imgs, origin, spacing, filename)

参考：Python中利用SimpleITK读取DICOM文件 - 简书