Python _get_center_kernel2d Examples

Example #1

File: test_spatial_softargmax.py Project: saurabhacellere/kornia

 def test_odd(self):
     kernel = _get_center_kernel2d(3, 3)
     expected = torch.tensor([[[[0., 0., 0.], [0., 1., 0.], [0., 0., 0.]],
                               [[0., 0., 0.], [0., 0., 0.], [0., 0., 0.]]],
                              [[[0., 0., 0.], [0., 0., 0.], [0., 0., 0.]],
                               [[0., 0., 0.], [0., 1., 0.], [0., 0., 0.]]]])
     assert_allclose(kernel, expected)

Example #2

 def test_even(self, device):
     kernel = _get_center_kernel2d(2, 2).to(device)
     expected = torch.ones(2, 2, 2, 2).to(device) * 0.25
     expected[0, 1] = 0
     expected[1, 0] = 0
     assert_allclose(kernel, expected)

Example #3

 def test_smoke(self, device):
     kernel = _get_center_kernel2d(3, 4).to(device)
     assert kernel.shape == (2, 2, 3, 4)

Example #4

File: test_spatial_softargmax.py Project: saurabhacellere/kornia

 def test_smoke(self):
     kernel = _get_center_kernel2d(3, 4)
     assert kernel.shape == (2, 2, 3, 4)