def imagevoi(input_datadir, output_filename):
'''
提取图像感兴趣的区域
'''
print("Hello")
reader = imagereader.vmtkImageReader()
# 图像序
filelist = os.listdir(input_datadir)
#print(len(filelist))
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()
selector = imagevoiselector.vmtkImageVOISelector()
selector.Image = reader.Image
selector.Execute()
writer = imagewriter.vmtkImageWriter()
writer.Image = selector.Image
writer.Format = "vtkxml"
writer.OutputFileName = output_filename
writer.WindowLevel = [150, 1200] #窗宽和窗位
writer.Execute()
def test_voi_selector(aorta_image, image_to_sha):
selector = voiselector.vmtkImageVOISelector()
selector.Image = aorta_image
selector.Interactive = 0
selector.BoxBounds = [157.0, 186.0, 31.0, 67.0, 10.0, 20.0]
selector.Execute()
assert image_to_sha(selector.Image) == 'f834f6ba5a9e8caf86fb9e0f951650e5adc627cc'
def test_voi_selector(aorta_image, compare_images):
name = __name__ + '_test_voi_selector.mha'
selector = voiselector.vmtkImageVOISelector()
selector.Image = aorta_image
selector.Interactive = 0
selector.BoxBounds = [157.0, 186.0, 31.0, 67.0, 10.0, 20.0]
selector.Execute()
assert compare_images(selector.Image, name) == True