(self,x)
| 19 | return [F.pad(x, self._pad, mode='replicate') for x in inputs] |
| 20 | |
| 21 | def unpad(self,x): |
| 22 | ht, wd = x.shape[-2:] |
| 23 | c = [self._pad[2], ht-self._pad[3], self._pad[0], wd-self._pad[1]] |
| 24 | return x[..., c[0]:c[1], c[2]:c[3]] |
| 25 | |
| 26 | def pytorch_foward_interpolate(flow): |
| 27 | def get_gaussian_weights(x, y, x1, x2, y1, y2): |