live_demo.py in PyTorch?

Open weirenorweiren opened this issue 1 year ago • 2 comments

Is it possible to provide a live_demo.py in PyTorch as you did for Colab demos? It could help the community take advantage of TAPIR in broader settings :)

Jul 09 '24 01:07 weirenorweiren

Hi @weirenorweiren , upon your request, we have added the pytorch live demo to our codebase.

Dec 02 '24 17:12 yangyi02

Hi @yangyi02, have the team done any performance test for the live demo written in PyTorch as the team did for JAX version?

In our tests, we achieved ~17 fps on 480x480 images on a quadro RTX 4000 (a 2018 mobile GPU).

Jan 21 '25 05:01 weirenorweiren

Hi @weirenorweiren , we have not conduct any performance test for the pytorch live demo, presumably the runtime shall be relatively the same under the same hardware. Feel free to give it a try and let us know how it works.

I will close this issue now. Feel free to reopen if you have further questions. Thanks.

Sep 24 '25 10:09 yangyi02