Method __init__

src/sobel.py:27–33

(self)

class Laplacian(nn.Module):
    def __init__(self):
        super().__init__()
        self.filter = nn.Conv2d(in_channels=3, out_channels=3, kernel_size=3, stride=1, padding=1, bias=False, groups=3)
        G = torch.tensor([[-1, -1, -1], [-1, 8, -1], [-1, -1, -1]]).float()
        G = G.unsqueeze(0).unsqueeze(0)
        G = torch.cat([G, G, G], 0)
        self.filter.weight = nn.Parameter(G, requires_grad=False)

    def forward(self, img):
        x = self.filter(img)