VTK绘制三维图像

一、程序框架

  基于VTK开发包，编写了一个用于显示三维图像的函数，主要用于调试过程中显示三维医学图像的语义信息。流程图如图1-1所示，代码详见github仓库，给个star再走哩！
github仓库链接     HappySleeepy的仓库

二、使用方式

       def plot_3D(_img, setting_color='silver', Spacing=None)

  函数有三个输入参数：_img、setting_color和Spacing

参数

_img: 用于绘制的三维数据。注意，输入要求是numpy数组，数据类型为np.uint8，取值为255或0（表示前景或背景）

setting_color : 设置前景颜色，默认为"silver"，可设置为"red"、“green"和"blue”

Spacing:设置原始图像采样间距，默认为None，当不为None时，会对原图像进行重采样，将采样间距统一为[1.0, 1.0, 1.0]. 注意，要求输入数据为numpy数据

环境要求

numpy

scipy

Simpleitk

vtk

   安装

                      pip install Plot-3D-Medical-Image

   使用实例

from Medical_Image_Plot import plot_3D
import numpy as np

# 自定义几何体
image = np.zeros((64, 64, 64), dtype=np.uint8)
for i in range(64):
    image[i, i:64-i, i:64-i]=255
# 调用库绘制三维图像
plot_3D(image)

   效果图

图2-1 3D绘制效果图

参考资料

1: MedicalDemo1
2: VTKWithNumpy
3: 发布自定义Python开发包