def __init__(self, args, image_size=None, do_augument=False):
self.image_size = image_size
self.do_augument = do_augument
if self.do_augument:
self.augumentor = FlowAugmentor(self.image_size)
self.flow_list = []
self.image_list = []
self.init_seed = False
def __init__(self, aug_params=None, sparse=False):
self.augmentor = None
self.sparse = sparse
if aug_params is not None:
if sparse:
self.augmentor = SparseFlowAugmentor(**aug_params)
else:
self.augmentor = FlowAugmentor(**aug_params)
self.augmentor.eraser_aug_prob = 0
self.is_test = False
self.init_seed = False
self.flow_list = []
self.image_list = []
def __init__(self,
args,
image_size=None,
do_augument=False,
return_path=False,
return_img_path=False):
self.image_size = image_size
self.do_augument = do_augument
self.return_path = return_path
self.return_img_path = return_img_path
if self.do_augument:
self.augumentor = FlowAugmentor(self.image_size)
self.flow_list = []
self.image_list = []