Vectorizing `amass_to_pose` in `raw_pose_processing`

EricGuo5513 / HumanML3D

HumanML3D: A large and diverse 3d human motion-language dataset.

https://ericguo5513.github.io/text-to-motion/

MIT License

829 stars 81 forks source link

Vectorizing `amass_to_pose` in `raw_pose_processing` #106

Open jojo978 opened 1 year ago

jojo978 commented 1 year ago

Hello,

I think amass_to_pose can be vectorized (not iterating a single pose at a time but all poses in a sequence (a npz file) at a single time.).

I tried this and it gave about >10x boost in speed. (since my graphics card GTX 1660 Ti gave me a CUDA memory allocation error, I calculated in cpu though.).

If it's not a resolved issue, I may issue a PR.

Thanks,

Young

sridharkamath commented 3 months ago

Hi I am facing a similar issue while processing (amass_to_pose) KIT dataset:

Can you please help out

sridharkamath commented 3 months ago

My GPU is 2080 Ti