dataset fix

2023-05-05 14:01:04 +00:00
parent d63c778ccd
commit 89d887e220
1 changed files with 1 additions and 1 deletions
@@ -159,7 +159,7 @@ class Dataset(Dataset):
        if 'frame_nr' in self.signature:
            item['frame_nr'] = frame.frame_nr
        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
            item['pose_shape'] = pose_shape
        if 'mesh_body_posed' in self.signature: