Blury on rotated image.

Here is a comparison between PIL rotation and kornia rotation. It is pretty clear that the image was blurred.

import torch

import kornia.augmentation.functional as F
from torchvision.transforms import functional as tvF
from torchvision.transforms import transforms

import matplotlib.pyplot as plt
import numpy as np

to_tensor = transforms.ToTensor()
to_pil = transforms.ToPILImage()

def tensor_pre_transform_wrapper(input: torch.Tensor):
    """ A wrapper that tried to reproduce the actual output from:
        - transforms.ToPILImage()
        - transforms.ToTensor()
        For each image, simply (img * 255).int() // 255
    """
    return (input * 255).int().float() / 255

in_tensor = torch.rand((3, 23, 44))
in_pil = to_pil(in_tensor)
degrees = 10
out_tensor = F._apply_rotation(
    tensor_pre_transform_wrapper(in_tensor), {'degrees': torch.tensor(degrees)}, False)
out_pil = tvF.rotate(in_pil, angle=degrees)

plt.imshow(np.moveaxis(out_tensor.numpy(), 0, -1))
plt.imshow(np.array(out_pil))

[edgarriba]

Let’s add that flag

[edgarriba]

@doube1008 to rotate image use kornia.geometry.transform.rotate