vchoutas / smplify-x

Expressive Body Capture: 3D Hands, Face, and Body from a Single Image
https://smpl-x.is.tue.mpg.de/
Other
1.73k stars 336 forks source link

Is it normal to take ~1 minute to fit one image? #35

Closed Lotayou closed 4 years ago

Lotayou commented 5 years ago

@vchoutas Hi it's Lotayou again, I disabled mesh intersection part and finally the fitting script runs okay. However it takes almost 1 min to fit a single image, as shown below: image

I'm wondering if this is the normal speed? For the record, as the script was running I checked the nvidia-smi and confirmed that the code was running on GPU. Is there anyway to speedup? Thanks! image

vchoutas commented 5 years ago

@Lotayou yes that is just about right. Unfortunately, I don't have any suggestions for improving the perfomance.

xiaobaozi1996 commented 4 years ago

Hi, I'm confused about which optimation step cause the fitting process takes extensive time consumption? Could you give me some advice?

Lotayou commented 4 years ago

@maoyali You can see the above profiler image for yourself. The most time-consuming part seems to be stage 004

xiaobaozi1996 commented 4 years ago

Oh, thanks for your answer. I have read the fit_smplx.yaml and found the last stage optimize all the loss. which component of loss is the most time-consuming?Do you have explored the problem or have some idea?