kornia fix

eps696 · Oct 28, 2021 · 22201e8 · 22201e8
1 parent 6830303
commit 22201e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transforms.py b/transforms.py
@@ -59,7 +59,7 @@ def inner(image_t):
         center[..., 1] = (image_t.shape[2] - 1) / 2
         try:
             M = kornia.geometry.transform.get_rotation_matrix2d(center, angle, scale).to(device)
-            rotated_image = kornia.geometry.transform..warp_affine(image_t.float(), M, dsize=(h, w))
+            rotated_image = kornia.geometry.transform.warp_affine(image_t.float(), M, dsize=(h, w))
         except:
             M = kornia.get_rotation_matrix2d(center, angle, scale).to(device)
             rotated_image = kornia.warp_affine(image_t.float(), M, dsize=(h, w))