def __init__(self, image_pipeline: Chain, mask_pipeline: Chain): store_attr()
def __init__(self, aug_transforms: List[BaseTransform]): store_attr() self.aug_transform_parameters = list(itertools.product(*[t.params for t in self.aug_transforms])) self.deaug_transforms = aug_transforms[::-1] self.deaug_transform_parameters = [p[::-1] for p in self.aug_transform_parameters]
def __init__(self, pname: str, params: Union[list, tuple]): store_attr() class Chain: