search

open-mmlab / mmhuman3d

OpenMMLab 3D Human Parametric Model Toolbox and Benchmark
https://mmhuman3d.readthedocs.io/
Apache License 2.0
1.23k stars 136 forks source link

_fits.npy for own dataset training SPIN model #106

Open JackieXuu opened 2 years ago

JackieXuu commented 2 years ago

I use my own dataset to train SPIN model and I found it need _fits.npy file. The train datasets in fits_dict.py are fixed and maybe if you can add a flag for other datasets and generate initial fits_npy file automatically?

caizhongang commented 2 years ago

@ZhengtianXu Thank you for the suggestion! fits_dict.py will be refactored shortly, and we will add tools to generate initial dict for SPIN (along with some other features such as feature extraction for VIBE).

Would you mind adding this request to the wishlist please?

JackieXuu commented 2 years ago

@ZhengtianXu Thank you for the suggestion! fits_dict.py will be refactored shortly, and we will add tools to generate initial dict for SPIN (along with some other features such as feature extraction for VIBE).

Would you mind adding this request to the wishlist please?

Yep, thanks for your help!