def test_pipeline(path):
image = read_image(path)
args = Compose([
Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
ChannelsFirst()
])(image=image)
return args['image']
def test_classification_pipeline(path):
image = read_image(path)
args = Compose([
Resize(224, 224),
Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
ChannelsFirst()
])(image=image)
return {'image': args['image']}
def read_image_and_mask(mask_db, path):
image = read_image(path)
mask = load_mask(mask_db, image.shape[:2], path)
args = Crop(0, 0, 256, 256)(image=image, mask=mask)
return args['image'], args.get('mask')
def read_image_and_mask(mask_db, path):
image = read_image(path)
mask = load_mask(mask_db, image.shape[:2], path)
return image, mask