Update transform.py

2023-03-23 18:55:24 +08:00
parent 3a752f5a26
commit 3aaf75021a
1 changed files with 1 additions and 1 deletions
@@ -63,7 +63,7 @@ class RandomHorizontalFlip(object):
        if random.uniform(0, 1) >= self.prob:
            return seq
        else:
-            return seq[:, :, ::-1]
+            return seq[..., ::-1]


 class RandomErasing(object):