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