Bart/bullet3
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

correct pickle loading

Browse Source
This commit is contained in:
Bart Moyaers
2019-08-08 16:03:48 +02:00
parent 05d86d5fd1
commit e499da08df
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/pybullet/gym/pybullet_envs/deep_mimic/mocap/h36m_dataset.py
View File

@@ -282,9 +282,18 @@ class Human36mDataset(MocapDataset):
cam['intrinsic'] = np.concatenate((cam['focal_length'], cam['center'], cam['intrinsic'] = np.concatenate((cam['focal_length'], cam['center'],
cam['radial_distortion'], cam['tangential_distortion'])) cam['radial_distortion'], cam['tangential_distortion']))
# save np.load
np_load_old = np.load
# modify the default parameters of np.load
np.load = lambda *a,**k: np_load_old(*a, allow_pickle=True, **k)
# Load serialized dataset # Load serialized dataset
data = np.load(path)['positions_3d'].item() data = np.load(path)['positions_3d'].item()
# restore np.load for future normal usage
np.load = np_load_old
self._data = {} self._data = {}
for subject, actions in data.items(): for subject, actions in data.items():
self._data[subject] = {} self._data[subject] = {}