diff --git a/opengait/data/transform.py b/opengait/data/transform.py index 8d57f73..61b362e 100644 --- a/opengait/data/transform.py +++ b/opengait/data/transform.py @@ -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):