it seems the inference speed is slow!

hellojialee / Improved-Body-Parts

Simple Pose: Rethinking and Improving a Bottom-up Approach for Multi-Person Pose Estimation

https://arxiv.org/abs/1911.10529

258 stars 42 forks source link

it seems the inference speed is slow! #10

Closed visonpon closed 4 years ago

visonpon commented 4 years ago

Hi, @jialee93 Thanks for your great work. When I use demo_images.py to test on my own image , each image consumes about 40 seconds(TitianV)!. But as you said in the readme, the speed has achived real time. I don't know why and hope you can give some advice~

hellojialee commented 4 years ago

Hi, first please check your environment configurations again. The speed is tested using pure python and in single process, please refer to the testing details in the readme. The postprocessing procedure can be improved and it has not been well done for release. Note that the keypoint assignment is based on greedy connection, which can be parallelly done in several independent processes.

hellojialee commented 4 years ago

For now, you can refer to this issue #20: https://github.com/hellojialee/Improved-Body-Parts/issues/20#issuecomment-633406270 I'm doing a further work and it really needs more time.