def __init__(self, dataset, crop='', scale='', distort='', norm=norm_RGB, what='aflow mask', idx_as_rng_seed=False): assert hasattr(dataset, 'npairs') assert hasattr(dataset, 'get_pair') self.dataset = dataset self.distort = instanciate_transformation(distort) self.crop = instanciate_transformation(crop) self.norm = instanciate_transformation(norm) self.scale = instanciate_transformation(scale) self.idx_as_rng_seed = idx_as_rng_seed self.what = what.split() if isinstance(what, str) else what self.n_samples = 5
def __init__(self, dataset, scale='', distort=''): PairDataset.__init__(self) assert isinstance(dataset, Dataset) and not isinstance(dataset, PairDataset) self.dataset = dataset self.npairs = dataset.nimg self.get_image = dataset.get_image self.get_key = dataset.get_key self.get_filename = dataset.get_filename self.root = None self.distort = instanciate_transformation(distort) self.scale = instanciate_transformation(scale)
def __init__(self, dataset, trf=''): super().__init__() self.dataset = dataset self.nimg = dataset.nimg self.npairs = dataset.npairs self.get_image = dataset.get_image self.get_key = dataset.get_key self.get_filename = dataset.get_filename self.root = None self.trf = instanciate_transformation(trf)
def __init__(self, dataset, crop='', scale='', distort='', norm=norm_RGB, what='aflow mask', do_color_aug=False, idx_as_rng_seed=False): assert hasattr(dataset, 'npairs') assert hasattr(dataset, 'get_pair') self.dataset = dataset self.distort = instanciate_transformation(distort) self.crop = instanciate_transformation(crop) self.norm = instanciate_transformation(norm) self.scale = instanciate_transformation(scale) self.idx_as_rng_seed = idx_as_rng_seed # to remove randomness self.what = what.split() if isinstance(what, str) else what self.n_samples = 5 # number of random trials per image self.DO_COLOR_AUG = do_color_aug self.aug = init_transforms(self.DO_COLOR_AUG)[0]
def __init__(self, dataset, trf=''): self.attach_dataset(dataset) self.trf = instanciate_transformation(trf)
def __init__(self, dataset, scale='', distort=''): self.attach_dataset(dataset) self.distort = instanciate_transformation(distort) self.scale = instanciate_transformation(scale)