Ejemplos de get_motion_kernel2d en Python

Ejemplo n.º 1

Archivo: test_motion.py Proyecto: zeta1999/kornia

def test_get_motion_kernel2d(batch_size, ksize, angle, direction):
    if batch_size != 0:
        angle = torch.tensor([angle] * batch_size)
        direction = torch.tensor([direction] * batch_size)
    else:
        batch_size = 1
    kernel = kornia.get_motion_kernel2d(ksize, angle, direction)
    assert kernel.shape == (batch_size, ksize, ksize)
    assert_allclose(kernel.sum(), batch_size)

Ejemplo n.º 2

Archivo: test_motion.py Proyecto: manyids2/kornia-1

def test_get_motion_kernel2d(ksize, angle, direction):
    kernel = kornia.get_motion_kernel2d(ksize, angle, direction)
    assert kernel.shape == (ksize, ksize)
    assert_allclose(kernel.sum(), 1.)