### 将 NII 文件按切片转换为 PNG 格式的 Python 实现 为了将 `.nii` 格式的医学图像按照每个切片逐一保存为 `.png` 图像，可以基于 VTK 库扩展功能。以下是详细的解决方案： #### 修改后的代码逻辑 通过调整原始代码中的部分逻辑[^1]，使程序能够读取 `.nii` 文件并将其逐一切片保存为单独的 `.png` 文件。 ```python import os import vtk def convert_nii_slices_to_png(input_folder, output_folder): # 获取输入文件夹中所有NII文件的路径 nii_files = [file for file in os.listdir(input_folder) if file.endswith('.nii')] # 遍历每一个NII文件 for nii_file in nii_files: # 构建输入文件路径 input_path = os.path.join(input_folder, nii_file) # 创建NII阅读器 nii_reader = vtk.vtkNIFTIImageReader() nii_reader.SetFileName(input_path) nii_reader.Update() # 获取数据维度信息 dimensions = nii_reader.GetOutput().GetDimensions() depth = dimensions[2] # 切片数量 # 对每一层切片进行处理 for slice_index in range(depth): # 提取当前切片的数据 extract_filter = vtk.vtkImageExtractComponents() extract_filter.SetInputConnection(nii_reader.GetOutputPort()) extract_filter.SetComponents(slice_index % 3) # 基于通道提取特定分量 extract_filter.Update() # 转换到PNG格式 png_writer = vtk.vtkPNGWriter() output_filename = f"{os.path.splitext(nii_file)[0]}_slice_{slice_index}.png" output_path = os.path.join(output_folder, output_filename) png_writer.SetFileName(output_path) png_writer.SetInputData(extract_filter.GetOutput()) # 执行写入操作 png_writer.Write() print(f"已成功将 {nii_file} 的所有切片转换为 PNG") print("批量转换完成！") # 设置输入文件夹和输出文件夹的路径 input_folder = '/path/to/input_folder' output_folder = '/path/to/output_folder' # 如果目标目录不存在，则创建它 if not os.path.exists(output_folder): os.makedirs(output_folder) # 执行批量转换 convert_nii_slices_to_png(input_folder, output_folder) ``` #### 关键点说明 - **切片索引**：利用 `vtkImageExtractComponents` 或其他过滤器来获取三维体积数据的不同二维切面。 - **动态命名机制**：对于每张图片，采用 `_slice_<index>` 形式作为后缀名以便区分不同切片[^1]。 - **错误检测与日志记录**：增加打印语句用于跟踪进度以及可能发生的异常情况。 此方法不仅实现了单个`.nii`文件向多张`.png`图转变的功能需求，还保持了较高的灵活性供进一步定制化开发之需。