Method __init__

core/utils/augmentor.py:7–13

(self, args)

class Augmentor:
    def __init__(self, args):

        H, W = args.image_size

        self.tnf_list = ['affine', 'hom', 'tps']
        self.tnf_generators = {tnf: SynthPairTnf(geometric_model=tnf, output_size=(H, W), use_cuda=False) for tnf in self.tnf_list}
        self.theta_generator = {tnf: Theta_gen(geometric_model=tnf, output_size=(H, W)) for tnf in self.tnf_list}

    def __call__(self, im, tnf_type, theta=None):
        '''