add shuffle

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