Vegetebird / MHFormer

[CVPR 2022] MHFormer: Multi-Hypothesis Transformer for 3D Human Pose Estimation
MIT License
530 stars 85 forks source link

Add Docker environment and web demo #50

Open vccheng2001 opened 2 years ago

vccheng2001 commented 2 years ago

Hey @Vegetebird ! 👋

This pull request makes it possible to run your model inside a Docker environment, which makes it easier for other people to run it. We're using an open source tool called Cog to make this process easier. To make the model work in Cog, a few edits have been made (for example, parsing args from command line using argparser wouldn't work so we manually set the default args in set_default_options() whenever use_cog= True).

This also means we can make a web demo where other people can try out your model! View it here: https://replicate.com/vegetebird/human-pose-estimation

We've added some examples to the web demo; please click the black "Claim this model" button andclaim your page here so you own it/edit it.

In case you're wondering who I am, I'm from Replicate, where we're trying to make machine learning reproducible. We got frustrated that we couldn't run all the really interesting ML work being done. So, we're going round implementing models we like. 😊

RohaanA commented 1 year ago

Wow this is very useful!