initialneil's picture
Upload folder using huggingface_hub
a7ff1cf verified
import pickle
class Struct(object):
def __init__(self, **kwargs):
for key, val in kwargs.items():
setattr(self, key, val)
with open(f'./FLAME_masks.pkl', 'rb') as f:
ss = pickle.load(f, encoding='latin1')
masks = Struct(**ss)