Removed dependency on Kornia library

jac99 · Mar 6, 2022 · 9e61877 · 9e61877
1 parent 18ed80b
commit 9e61877
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data/augmentation.py b/data/augmentation.py
@@ -211,7 +211,7 @@ def get_params(self, h, w):
             else:
                 assert NotImplementedError('Incorrect shear: {}'.format(self.shear))
         else:
-            shear = 0.0
+            shear = [0., 0.]
 
         return angle, translations, scale, shear