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