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

dataset fix

Browse Source
This commit is contained in:
luni
2023-05-05 14:01:04 +00:00
parent d63c778ccd
commit 89d887e220
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/dataset.py
+1 -1
View File
@@ -159,7 +159,7 @@ class Dataset(Dataset):
if 'frame_nr' in self.signature: if 'frame_nr' in self.signature:
item['frame_nr'] = frame.frame_nr item['frame_nr'] = frame.frame_nr
if 'pose_shape' in self.signature: if 'pose_shape' in self.signature:
pose_shape = np.load(os.path.join(self.folder, 'pose_shape', frame.id + separator + frame.frame_nr + '.npy')) pose_shape = np.load(os.path.join(self.folder, 'pose_shape', frame.id + separator + frame.frame_nr + '.npy')).astype(np.float32)
pose_shape[:3] = 0 pose_shape[:3] = 0
item['pose_shape'] = pose_shape item['pose_shape'] = pose_shape
if 'mesh_body_posed' in self.signature: if 'mesh_body_posed' in self.signature: