提交 11e90ad9 编写于 作者: D Dario Pavllo

Fix data loading on NumPy 1.16.3

上级 1551a01d
......@@ -231,7 +231,7 @@ class Human36mDataset(MocapDataset):
cam['tangential_distortion']))
# Load serialized dataset
data = np.load(path)['positions_3d'].item()
data = np.load(path, allow_pickle=True)['positions_3d'].item()
self._data = {}
for subject, actions in data.items():
......
......@@ -108,7 +108,7 @@ class HumanEvaDataset(MocapDataset):
self._cameras[prefix + subject] = data
# Load serialized dataset
data = np.load(path)['positions_3d'].item()
data = np.load(path, allow_pickle=True)['positions_3d'].item()
self._data = {}
for subject, actions in data.items():
......
......@@ -58,7 +58,7 @@ for subject in dataset.subjects():
anim['positions_3d'] = positions_3d
print('Loading 2D detections...')
keypoints = np.load('data/data_2d_' + args.dataset + '_' + args.keypoints + '.npz')
keypoints = np.load('data/data_2d_' + args.dataset + '_' + args.keypoints + '.npz', allow_pickle=True)
keypoints_symmetry = keypoints['metadata'].item()['keypoints_symmetry']
kps_left, kps_right = list(keypoints_symmetry[0]), list(keypoints_symmetry[1])
joints_left, joints_right = list(dataset.skeleton().joints_left()), list(dataset.skeleton().joints_right())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册