(self, dims, mode='sintel')
| 7 | class InputPadder: |
| 8 | """ Pads images such that dimensions are divisible by 8 """ |
| 9 | def __init__(self, dims, mode='sintel'): |
| 10 | self.ht, self.wd = dims[-2:] |
| 11 | pad_ht = (((self.ht // 8) + 1) * 8 - self.ht) % 8 |
| 12 | pad_wd = (((self.wd // 8) + 1) * 8 - self.wd) % 8 |
| 13 | if mode == 'sintel': |
| 14 | self._pad = [pad_wd//2, pad_wd - pad_wd//2, pad_ht//2, pad_ht - pad_ht//2] |
| 15 | elif mode == 'kitti400': |
| 16 | self._pad = [0, 0, 0, 400 - self.ht] |
| 17 | else: |
| 18 | self._pad = [pad_wd//2, pad_wd - pad_wd//2, 0, pad_ht] |
| 19 | |
| 20 | def pad(self, *inputs): |
| 21 | return [F.pad(x, self._pad, mode='replicate') for x in inputs] |
nothing calls this directly
no outgoing calls
no test coverage detected