def create_generators()

in augerino_lib/uniform_aug.py [0:0]

• 44 lines of code
• 7 McCabe index (conditional complexity)

    def create_generators(self, bs):
        g = []
        ## tx
        if 'tx' in self.transfos:
            print("Using tx")
            gx = torch.zeros(3, 3)
            gx[0, 2] = 1. 
            gx = gx.unsqueeze(-1).unsqueeze(-1).expand(3,3, bs, 1)
            g.append(gx)

        ## ty
        if 'ty' in self.transfos:
            print("Using ty")
            gy = torch.zeros(3, 3)
            gy[1, 2] = 1. 
            gy = gy.unsqueeze(-1).unsqueeze(-1).expand(3,3, bs, 1)
            g.append(gy)

        if 'rot' in self.transfos:
            print("Using rot")
            gr = torch.zeros(3, 3)
            gr[0, 1] = -1.
            gr[1, 0] = 1.
            gr = gr.unsqueeze(-1).unsqueeze(-1).expand(3,3, bs, 1)
            g.append(gr)

        if 'scale' in self.transfos:
            print("Using scale")
            gs = torch.zeros(3, 3) 
            gs[0, 0] = 1. 
            gs[1, 1] = 1. 
            gs = gs.unsqueeze(-1).unsqueeze(-1).expand(3,3, bs, 1)
            g.append(gs)

        if 'strech' in self.transfos:
            print("Using strech")
            gst = torch.zeros(3, 3)
            gst[0, 0] = 1.
            gst[1, 1] = -1.
            gst = gst.unsqueeze(-1).unsqueeze(-1).expand(3,3, bs, 1)
            g.append(gst)

        if 'shear' in self.transfos:
            print("Using shear")
            gsh = torch.zeros(3, 3) 
            gsh[0, 1] = 1.
            gsh[1, 0] = 1.
            gsh = gsh.unsqueeze(-1).unsqueeze(-1).expand(3,3, bs, 1)
            g.append(gsh)
            
        g_matrices = torch.cat(g, dim = -1)
        self.g_matrices = nn.Parameter(data=g_matrices, requires_grad=False)