Walter0807 / MotionBERT

[ICCV 2023] PyTorch Implementation of "MotionBERT: A Unified Perspective on Learning Human Motion Representations"
Apache License 2.0
1.02k stars 124 forks source link

Real time application #29

Closed blazapropeler closed 1 year ago

blazapropeler commented 1 year ago

Hi!

I was just wondering if you have some results on the speed and if this model (in the Lite variant) would be suitable for a real-time 3d pose estimation problem?

Thanks

Walter0807 commented 1 year ago

We have not tested the speed, but from my experience the inference is very fast with GPU. It seems like real-time works from this issue.

lucasjinreal commented 1 year ago

I have tested the pose3d, it was extremly fast. However I found the mesh3d model ran twice in infer_wild_mesh file. @Walter0807 Do u know how much effect loose if we just using single forward on mesh3d inference?

Walter0807 commented 1 year ago

I have tested the pose3d, it was extremly fast. However I found the mesh3d model ran twice in infer_wild_mesh file. @Walter0807 Do u know how much effect loose if we just using single forward on mesh3d inference?

You can remove the flipping part if speed is important to you, it would slightly increase the error from my experience. You might refer to this script.

XALEX-123 commented 1 year ago

hello @lucasjinreal, I recently try to inference motionbert in real time, but i'm stuck. Now I can use AlphaPose to do 2D extraction in real time, however, i don't know how to use this real time 2D extraction as an input to the motionbert, can you guide me or give me some advice? thank you so much!

lucasjinreal commented 1 year ago

@XALEX-123 Actually I found motionBert has large performance reduce if you are not sampling on whole sequence, you should try on python first, take a deep look at performance in realtime situation.

XALEX-123 commented 1 year ago

@lucasjinreal thank your rapid reply, then how can i inference motionbert in real time?

luoww1992 commented 1 year ago

@lucasjinreal can your share your main env and GPU on your PC, how many fps ?