From 89d887e22083030753fba6fac04b183ffb947a37 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Fri, 5 May 2023 14:01:04 +0000 Subject: [PATCH] dataset fix --- data/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/dataset.py b/data/dataset.py index 0a2db61..188f1b2 100644 --- a/data/dataset.py +++ b/data/dataset.py @@ -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: