Method __init__

core/utils/transformation.py:386–390

(self, out_h=240, out_w=240, out_ch=3, use_cuda=True)

class AffineGridGen(Module):
    def __init__(self, out_h=240, out_w=240, out_ch=3, use_cuda=True):
        super(AffineGridGen, self).__init__()
        self.out_h = out_h
        self.out_w = out_w
        self.out_ch = out_ch

    def forward(self, theta):
        b = theta.size()[0]