Python vtk格式转nii

最新推荐文章于 2024-07-16 15:46:29 发布

原创最新推荐文章于 2024-07-16 15:46:29 发布 · 708 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#python #图像处理

收录于

该文章介绍了一个使用Python和vtk库将vtk格式的图像数据批量转换为nii格式的脚本。通过vtkStructuredPointsReader读取vtk文件，然后用vtkNIFTIImageWriter保存为nii格式，实现文件的转换。脚本适用于同一目录下所有vtk文件。

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()