def levelset_seg():
'''
图像的水平集分割，生成等值面
'''
input_file = "vesselVoi.vti"
output_filename = 'vessellevelset.vtp'
reader = imagereader.vmtkImageReader()
# 图像序
reader.InputFileName = input_file
reader.Format = "vtkxml"
reader.UseITKIO = 0
reader.Execute()
levelset = levelsetseg.vmtkLevelSetSegmentation()
levelset.Image = reader.Image
levelset.LevelSetsType = "geodesic" # '["geodesic","curves","threshold","laplacian"]'
levelset.FeatureImageType = "gradient" # '["vtkgradient","gradient","upwind","fwhm"]'
levelset.Execute()
marchingcube = marchingcubes.vmtkMarchingCubes()
marchingcube.Image = levelset.LevelSetsOutput
marchingcube.Execute()
writer = surfacewriter.vmtkSurfaceWriter()
writer.Surface = marchingcube.Surface
writer.Format = "vtkxml"
writer.OutputFileName = output_filename
writer.Execute()
def test_marching_cubes_default(level_set_image, compare_surfaces):
name = __name__ + '_test_marching_cubes_default.vtp'
mc = marchingcubes.vmtkMarchingCubes()
mc.Image = level_set_image
mc.Level = 0.0
mc.Execute()
assert compare_surfaces(mc.Surface, name) == True
def test_marching_cubes_default(level_set_image, compare_surfaces):
name = __name__ + '_test_marching_cubes_default.vtp'
mc = marchingcubes.vmtkMarchingCubes()
mc.Image = level_set_image
mc.Level = 0.0
mc.Execute()
assert compare_surfaces(mc.Surface, name) == True
def test_default_parameters(transform_image, compare_surfaces):
name = __name__ + '_test_default_parameters.stl'
import vmtk.vmtkmarchingcubes as marching
image = transform_image[0]
matrixCoefficients = transform_image[1]
mc = marching.vmtkMarchingCubes()
mc.Image = image
mc.Level = 3000.0
mc.Execute()
transform = transformtoras.vmtkSurfaceTransformToRAS()
transform.Surface = mc.Surface
transform.XyzToRasMatrixCoefficients = matrixCoefficients
transform.Execute()
assert compare_surfaces(transform.Surface, name) == True
def test_default_parameters(transform_image, compare_surfaces):
name = __name__ + '_test_default_parameters.stl'
import vmtk.vmtkmarchingcubes as marching
image = transform_image[0]
matrixCoefficients = transform_image[1]
mc = marching.vmtkMarchingCubes()
mc.Image = image
mc.Level = 3000.0
mc.Execute()
transform = transformtoras.vmtkSurfaceTransformToRAS()
transform.Surface = mc.Surface
transform.XyzToRasMatrixCoefficients = matrixCoefficients
transform.Execute()
assert compare_surfaces(transform.Surface, name) == True