Closed Okaerikoto closed 2 years ago
Hmm, that also just confused me a little bit. I think there are both .pkl
and .npz
versions of SMPL-H. When you download [1] you can find the .pkl
versions in the models/
subfolder. This is how I got them. Note that you have to preprocess them according to the notes in the smplx project.
If you download [2] then you'll indeed get .npz
files. By default, we expect .pkl
files, because we essentially just wrap the smplx.create
function and that function expects .pkl
by default. However, it can also load .npz
files, so you should be able to load the .npz
version if you change SMPLSequence.from_amass
to load the SMPL layer as follows
smpl_layer = SMPLLayer(model_type='smplh', gender=body_data['gender'].item(), device=C.device, ext='npz')
Can you try if that works for you? If so, I can quickly push a change that exposes this parameter to the .from_amass
function.
Thanks for the answer, that worked (download the first version + using the script as pointed in notes in the smplx project. Cool demo, thanks for releasing this :).
I am trying to execute the script load_amass.py.
It expects .pkl models for SMPL-H, but the models SMPL-H for amass were released as .npz.
From MANO Download page, which models should I download, the first link or the second? [1] Models & Code [2] Extended SMPL+H model (used in AMASS project)
Whether I try to load the first version (.pkl) or the second version (.npz), I get the following error: