Python vtk格式转nii

vtk格式文件转nii格式

导入相关库

import vtk

安装方法

pip install vtk

批量操作目录下的vtk文件

import glob
import os
from pathlib import Path

文件地址

dir_path = 'vtk2nii/T2 yuanshi'

完整代码：

import vtk
import glob
import os
from pathlib import Path

dir_path = 'vtk2nii/T2 yuanshi'
path = Path(dir_path)
for file_path in path.glob('*'):
    # print(file_path)
    reader = vtk.vtkStructuredPointsReader()
    reader.SetFileName(file_path)
    reader.Update()
    image_data = reader.GetOutput()
    # 保存为nii.gz
    writer = vtk.vtkNIFTIImageWriter()
    writer.SetInputData(image_data)
    file_name = file_path.stem
    # print(file_name)
    writer.SetFileName(f'vtk2nii/T2 yuanshi_nii/{file_name}.nii')
    writer.Write()

转换完成