stack-3d/test/render_networks.py

11 lines
275 B
Python
Raw Normal View History

import vtk
import stack3d.formats.unet
path = "/home/grogan/stack-working/skeleton.pkl"
network_vtk = stack3d.formats.unet.skeleton_to_vtp(path)
writer = vtk.vtkXMLPolyDataWriter()
writer.SetInputData(network_vtk)
writer.SetFileName("/home/grogan/test.vtp")
writer.Write()