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