o €o™iã@s’ddlZddlmZddlmZgd¢ZGdd„deƒZGdd„deƒZGd d „d eƒZd ed efd d„Z d ed efdd„Z d ed efdd„Z dS)éN)Ú ImageModule)ÚTensor)ÚHflipÚRot180ÚVflipÚhflipÚrot180Úvflipc@ó0eZdZdZdedefdd„Zdefdd„ZdS) ra2Vertically flip a tensor image or a batch of tensor images. Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The vertically flipped image tensor. Examples: >>> vflip = Vflip() >>> input = torch.tensor([[[ ... [0., 0., 0.], ... [0., 0., 0.], ... [0., 1., 1.] ... ]]]) >>> vflip(input) tensor([[[[0., 1., 1.], [0., 0., 0.], [0., 0., 0.]]]]) ÚinputÚreturncCót|ƒS©N)r ©Úselfr ©rúS/home/ubuntu/.local/lib/python3.10/site-packages/kornia/geometry/transform/flips.pyÚforward3óz Vflip.forwardcCó|jjSr©Ú __class__Ú__name__©rrrrÚ__repr__6rzVflip.__repr__N©rÚ __module__Ú __qualname__Ú__doc__rrÚstrrrrrrrórc@r ) ra6Horizontally flip a tensor image or a batch of tensor images. Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The horizontally flipped image tensor. Examples: >>> hflip = Hflip() >>> input = torch.tensor([[[ ... [0., 0., 0.], ... [0., 0., 0.], ... [0., 1., 1.] ... ]]]) >>> hflip(input) tensor([[[[0., 0., 0.], [0., 0., 0.], [1., 1., 0.]]]]) r r cCr r)rrrrrrSrz Hflip.forwardcCrrrrrrrrVrzHflip.__repr__Nrrrrrr:r rc@r ) raýRotate a tensor image or a batch of tensor images 180 degrees. Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Examples: >>> rot180 = Rot180() >>> input = torch.tensor([[[ ... [0., 0., 0.], ... [0., 0., 0.], ... [0., 1., 1.] ... ]]]) >>> rot180(input) tensor([[[[1., 1., 0.], [0., 0., 0.], [0., 0., 0.]]]]) r r cCr r)rrrrrrprzRot180.forwardcCrrrrrrrrsrzRot180.__repr__NrrrrrrZsrr r cCst |ddg¡S)aRotate a tensor image or a batch of tensor images 180 degrees. .. image:: _static/img/rot180.png Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The rotated image tensor. éþÿÿÿéÿÿÿÿ)ÚtorchÚflip©r rrrrwsrcCó| d¡ ¡S)a)Horizontally flip a tensor image or a batch of tensor images. .. image:: _static/img/hflip.png Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The horizontally flipped image tensor. r"©r$Ú contiguousr%rrrrˆórcCr&)a%Vertically flip a tensor image or a batch of tensor images. .. image:: _static/img/vflip.png Input must be a tensor of shape (C, H, W) or a batch of tensors :math:`(*, C, H, W)`. Args: input: input tensor. Returns: The vertically flipped image tensor. r!r'r%rrrr ™r)r ) r#Ú kornia.corerÚModulerÚ__all__rrrrrr rrrrÚs