Image Gradients

Image

Image Gradients

Functional Interface

torchmetrics.functional.image_gradients(img)

Wrapper for deprecated import.

>>> import torch
>>> image = torch.arange(0, 1*1*5*5, dtype=torch.float32)
>>> image = torch.reshape(image, (1, 1, 5, 5))
>>> dy, dx = _image_gradients(image)
>>> dy[0, 0, :, :]
tensor([[5., 5., 5., 5., 5.],
        [5., 5., 5., 5., 5.],
        [5., 5., 5., 5., 5.],
:rtype: :py:data:`~typing.Tuple`\[:py:class:`~torch.Tensor`, :py:class:`~torch.Tensor`]

[5., 5., 5., 5., 5.], [0., 0., 0., 0., 0.]])