luni/reconstruction
1
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

add shuffle

Browse Source
This commit is contained in:
luni
2023-06-17 16:11:21 +00:00
parent 0b9daf4841
commit 37c67d5cab
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/pose_dataset.py
+2
View File
@@ -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 superSMPL.smpl_torch import SMPL
from data.pose_renderer import PoseRenderer from data.pose_renderer import PoseRenderer
import trimesh import trimesh
import random
class PoseDataset(Dataset): class PoseDataset(Dataset):
@@ -23,6 +24,7 @@ class PoseDataset(Dataset):
self.device = device self.device = device
self.ae_mode = ae_mode self.ae_mode = ae_mode
self.frames = read_frames_file(os.path.join(folder, frames_file)) self.frames = read_frames_file(os.path.join(folder, frames_file))
random.shuffle(self.frames)
self.reader = DataReader(folder, 'superSMPL') self.reader = DataReader(folder, 'superSMPL')
self.cache = dict() if use_cache else None self.cache = dict() if use_cache else None