Spaces:
Running
on
L40S
Running
on
L40S
| # Copyright (c) OpenMMLab. All rights reserved. | |
| import numpy as np | |
| import torch | |
| import torch.nn.functional as F | |
| class TestBilinearGridSample: | |
| def _test_bilinear_grid_sample(self, | |
| dtype=torch.float, | |
| align_corners=False, | |
| multiplier=1, | |
| precision=1e-3): | |
| from mmcv.ops.point_sample import bilinear_grid_sample | |
| input = torch.rand(1, 1, 20, 20, dtype=dtype) | |
| grid = torch.Tensor([[[1, 0, 0], [0, 1, 0]]]) | |
| grid = F.affine_grid( | |
| grid, (1, 1, 15, 15), align_corners=align_corners).type_as(input) | |
| grid *= multiplier | |
| out = bilinear_grid_sample(input, grid, align_corners=align_corners) | |
| ref_out = F.grid_sample(input, grid, align_corners=align_corners) | |
| assert np.allclose(out.data.detach().cpu().numpy(), | |
| ref_out.data.detach().cpu().numpy(), precision) | |
| def test_bilinear_grid_sample(self): | |
| self._test_bilinear_grid_sample(torch.double, False) | |
| self._test_bilinear_grid_sample(torch.double, True) | |
| self._test_bilinear_grid_sample(torch.float, False) | |
| self._test_bilinear_grid_sample(torch.float, True) | |
| self._test_bilinear_grid_sample(torch.float, False) | |
| self._test_bilinear_grid_sample(torch.float, True, 5) | |
| self._test_bilinear_grid_sample(torch.float, False, 10) | |
| self._test_bilinear_grid_sample(torch.float, True, -6) | |
| self._test_bilinear_grid_sample(torch.float, False, -10) | |
| self._test_bilinear_grid_sample(torch.double, True, 5) | |
| self._test_bilinear_grid_sample(torch.double, False, 10) | |
| self._test_bilinear_grid_sample(torch.double, True, -6) | |
| self._test_bilinear_grid_sample(torch.double, False, -10) | |