From 3aaf75021a7b48f790655f6d0d36189664097333 Mon Sep 17 00:00:00 2001
From: Chao Fan <92680886+ChaoFan996@users.noreply.github.com>
Date: Thu, 23 Mar 2023 18:55:24 +0800
Subject: [PATCH] Update transform.py

---
 opengait/data/transform.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):