From 37c67d5cabf0097822762e6150fa7f7ab54d85c1 Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Sat, 17 Jun 2023 16:11:21 +0000
Subject: [PATCH] add shuffle

---
 data/pose_dataset.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/data/pose_dataset.py b/data/pose_dataset.py
index 8aeaef9..caa581b 100644
--- a/data/pose_dataset.py
+++ b/data/pose_dataset.py
@@ -15,6 +15,7 @@ from utils.data_utils import Frame, read_frames_file, mask_faces, get_normals_fr
 from superSMPL.smpl_torch import SMPL
 from data.pose_renderer import PoseRenderer
 import trimesh
+import random
 
 
 class PoseDataset(Dataset):
@@ -23,6 +24,7 @@ class PoseDataset(Dataset):
         self.device = device
         self.ae_mode = ae_mode
         self.frames = read_frames_file(os.path.join(folder, frames_file))
+        random.shuffle(self.frames)
         self.reader = DataReader(folder, 'superSMPL')
         
         self.cache = dict() if use_cache else None