Fixed bug in img_lms_pose_transforms.RandomRotation

YuvalNirkin · May 3, 2020 · 6bbb7a9 · 6bbb7a9
1 parent 962398f
commit 6bbb7a9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/datasets/img_lms_pose_transforms.py b/datasets/img_lms_pose_transforms.py
@@ -489,6 +489,7 @@ def __call__(self, x, angle=None):
         if angle is None or self.randomize_per_image:
             angle = (random.random() * 2.0 - 1.0) * self.max_degrees
         if isinstance(x, (list, tuple)):
+            x = list(x) if isinstance(x, tuple) else x
             for i in range(len(x)):
                 if is_img(x[i]):
                     angle = (random.random() * 2.0 - 1.0) * self.max_degrees if self.randomize_per_image else angle