| 87 | |
| 88 | |
| 89 | class Downsample(nn.Module): |
| 90 | def __init__(self, kernel, factor=2): |
| 91 | super().__init__() |
| 92 | |
| 93 | self.factor = factor |
| 94 | kernel = make_kernel(kernel) |
| 95 | self.register_buffer('kernel', kernel) |
| 96 | |
| 97 | p = kernel.shape[0] - factor |
| 98 | |
| 99 | pad0 = (p + 1) // 2 |
| 100 | pad1 = p // 2 |
| 101 | |
| 102 | self.pad = (pad0, pad1) |
| 103 | |
| 104 | def forward(self, input): |
| 105 | return upfirdn2d(input, self.kernel, up=1, down=self.factor, pad=self.pad) |
| 106 | |
| 107 | |
| 108 | class Blur(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected