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)