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