Convert VTK image to ITK image without itk-vtkglue

dchen · April 27, 2020, 1:28pm

There’s a corresponding function itk.GetImageViewFromArray that converts a numpy array to an ITK image. Here’s a blog post about that:

To get from a VTK image to NumPy array you can use the vtk_to_numpy in numpy_support. Here’s stackoverflow thread describing that process:

I wrote a function that goes from VTK to SimpleITK that is essentially the same thing:

github.com

dave3d/dicom2stl/blob/master/utils/vtk2sitk.py

#!/usr/bin/env python

import SimpleITK as sitk
import numpy
import vtk
from vtk.util.numpy_support import *

def vtk2sitk(vtkimg, debug=False):
    """Takes a VTK image, returns a SimpleITK image."""
    sd = vtkimg.GetPointData().GetScalars()
    npdata = vtk_to_numpy(sd)

    dims = list(vtkimg.GetDimensions())
    origin = vtkimg.GetOrigin()
    spacing=vtkimg.GetSpacing()

    if debug:
        print("dims:", dims)
        print("origin:", origin)
        print("spacing:", spacing)

This file has been truncated. show original