dictionary data npy로 저장/불러오기

npy로 dictionary data가 저장된 경우 불러올 때의 코드입니다.

import numpy as np
params = {'beta': np.array(betas), 'pose': pose}
np_path = 'params.npy')
np.save(np_path, params)

저장하기

 

import numpy as np

params = np.load('params.npy', allow_pickle=True)
print(params.item().get('beta').shape)
print(params.item().get('pose').shape)

불러오기 : 불러올 때는 item().get('키') 형식으로 불러와서 사용하면 됩니다. 

 

참조 : https://stackoverflow.com/questions/40219946/python-save-dictionaries-through-numpy-save

Python : save dictionaries through numpy.save