From 6d4cbe885f35ccea660a59869bbb880acb1a1257 Mon Sep 17 00:00:00 2001 From: Vadim Epstein Date: Thu, 28 Oct 2021 19:24:45 +0300 Subject: [PATCH] kornia fix --- transforms.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/transforms.py b/transforms.py index 1209f1b..fb6e05d 100644 --- a/transforms.py +++ b/transforms.py @@ -58,10 +58,9 @@ def inner(image_t): center[..., 0] = (image_t.shape[3] - 1) / 2 center[..., 1] = (image_t.shape[2] - 1) / 2 try: - M = K.get_rotation_matrix2d(center, angle, scale).to(device) + M = kornia.geometry.transform.get_rotation_matrix2d(center, angle, scale).to(device) except: M = kornia.get_rotation_matrix2d(center, angle, scale).to(device) - M = kornia.get_rotation_matrix2d(center, angle, scale).to(device) rotated_image = kornia.warp_affine(image_t.float(), M, dsize=(h, w)) return rotated_image return inner