PHALP tracker taking lot of CPU utilization

brjathu / PHALP

Code repository for the paper "Tracking People by Predicting 3D Appearance, Location & Pose". (CVPR 2022 Oral)

Other

282 stars 43 forks source link

PHALP tracker taking lot of CPU utilization #21

Open 9e0 opened 10 months ago

9e0 commented 10 months ago

Hi,

I find that PHALP tracker taking lot of CPU utilization, as numpy does not support GPU. Enabled code to run on CUDA, still the CPU utilization is heavy.

I feel the performance might be better if numpy array transactions can be moved to GPU

Is there a way I can make the PHALP tracking to run on GPU

Thanks in advance