Я пытаюсь использовать функцию kornia.geometry.transform.rotate в Python, чтобы повернуть тензор PyTorch на произвольные углы. Однако, если я сделаю простой поворот на 90 градусов, результирующий тензор не будет выглядеть как полностью повернутый.
Вот пример кода:
import torch
from kornia.geometry.transform import rotate
import matplotlib.pyplot as plt
a = torch.ones((1,64,64))
a[0,:,2] += 1
angle = torch.tensor([90])
c = rotate(a,angle)
plt.figure()
plt.subplot(121)
plt.imshow(a[0].detach().numpy())
plt.subplot(122)
plt.imshow(c[0].detach().numpy())
И результаты до и после вращение:
Мне не хватает тонкости из-за слишком грубого тензора, который вызывает проблемы с интерполяцией или что-то, что может быть значительно смягчено мелкозернистый тензор?
Заранее большое спасибо!
Примечание Я использую:
python 3.6.10
корния 0.2.0
pytorch 1.4.0