bd21bb4fec8d10b257ceccc3d302abd6d5b7c7d5,kornia/filters/kernels.py,,get_motion_kernel2d,#Any#Any#Any#,447

Before Change

    if not isinstance(angle, float):
        raise TypeError("angle must be a float")

    if not isinstance(direction, float):
        raise TypeError("direction must be a float")

    kernel_tuple: Tuple[int, int] = (kernel_size, kernel_size)
    // direction from [-1, 1] to [0, 1] range
    direction = (torch.clamp(torch.tensor(direction), -1., 1.).item() + 1.) / 2.
    kernel = torch.zeros(kernel_tuple, dtype=torch.float)

After Change

    if not isinstance(angle, torch.Tensor):
        angle = torch.tensor(angle)

    assert angle.dim() == 0, f"angle must be a 0-dim tensor. Got {angle}."

    if not isinstance(direction, torch.Tensor):
        direction = torch.tensor(direction)

Instances