kornia fix

eps696 · Oct 28, 2021 · 6d4cbe8 · 6d4cbe8
1 parent 4acd1b6
commit 6d4cbe8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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