def imageObjectEnhance():
input_datadir = 'C:/Users/chenjiaxing/Desktop/CT-Data/png/WU_Lung_Vessel_Ext/'
reader = imagereader.vmtkImageReader()
# 图像序
filelist = os.listdir(input_datadir)
reader.InputFilePrefix = input_datadir
reader.InputFilePattern = "%s%d.png"
reader.DataExtent = [0, 512, 0, 512, 1, len(filelist)]
reader.Format = "png"
reader.UseITKIO = 0
reader.Execute()
imageObjEn = imageEnhance.vmtkImageObjectEnhancement()
imageObjEn.Image = reader.Image
# default arguments
imageObjEn.SigmaMin = 1.0
imageObjEn.SigmaMax = 1.0
imageObjEn.NumberOfSigmaSteps = 1
imageObjEn.ScaledObjectness = 0
imageObjEn.Alpha = 0.5
imageObjEn.Beta = 0.5
imageObjEn.Gamma = 5.0
imageObjEn.ObjectDimension = 2
imageObjEn.Execute()
viewer = imageviewer.vmtkImageViewer()
viewer.Image = imageObjEn.Image
viewer.Execute()
def test_enhance_image_with_defaults(aorta_image, compare_images):
name = __name__ + '_test_enhance_image_with_defaults.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_with_defaults(aorta_image, compare_images):
name = __name__ + '_test_enhance_image_with_defaults.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_three_sigma_steps(aorta_image, compare_images):
name = __name__ + '_test_enhance_image_three_sigma_steps.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.NumberOfSigmaSteps = 3
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_with_scaled_objectiveness(aorta_image, compare_images):
name = __name__ + '_test_enhance_image_with_scaled_objectiveness.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.ScaledObjectness = True
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_three_sigma_steps(aorta_image, compare_images):
name = __name__ + '_test_enhance_image_three_sigma_steps.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.NumberOfSigmaSteps = 3
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_with_scaled_objectiveness(aorta_image, compare_images):
name = __name__ + '_test_enhance_image_with_scaled_objectiveness.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.ScaledObjectness = True
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_with_changing_sigma(aorta_image, compare_images, sigma_min, sigma_max, paramid):
name = __name__ + '_test_enhance_image_with_changing_sigma_' + paramid + '.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.SigmaMin = sigma_min
enhancer.SigmaMax = sigma_max
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_with_changing_sigma(aorta_image, compare_images,
sigma_min, sigma_max, paramid):
name = __name__ + '_test_enhance_image_with_changing_sigma_' + paramid + '.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.SigmaMin = sigma_min
enhancer.SigmaMax = sigma_max
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_with_changing_abg(aorta_image, compare_images, alpha, beta, gamma, paramid):
name = __name__ + '_test_enhance_image_with_changing_abg_' + paramid + '.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.Alpha = alpha
enhancer.Beta = beta
enhancer.Gamma = gamma
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
def test_enhance_image_with_changing_abg(aorta_image, compare_images, alpha,
beta, gamma, paramid):
name = __name__ + '_test_enhance_image_with_changing_abg_' + paramid + '.mha'
enhancer = enhance.vmtkImageObjectEnhancement()
enhancer.Image = aorta_image
enhancer.Alpha = alpha
enhancer.Beta = beta
enhancer.Gamma = gamma
enhancer.Execute()
assert compare_images(enhancer.Image, name) == True
