WholeBody models

[oscarfossey]

Hi, thank you for your work!
Do you know if the current implementation would work with the large wholdebody_model from Vit_pose?

[gpastal24]

Hello,
I am not sure I can find pretrained weights to test it. Nonetheless, I will make some changes to the repo.
The package version works with only vitbpose b right now.

[oscarfossey]

Alright it's clear. If you need them the large wholebody weights here : large_wholebody
The weight's link are at the bottom of the readme of the official repo. In the "WholeBody" part. For small, base, large and huge models. It says that you have to split them with their script it works well. But using mmpose implies to have an mmcv dependencie 🙅 .

[gpastal24]

Hi ok I thought vitpose+ was a different architecture. I will look into this the coming weekend and let you know . I ll probably need to find/write some visualization functions for this case as well.

[keniel123]

Hi
By chance any update on this ?

[gpastal24]

Hi, sorry for the late reply. I had dropped this project, since I had to focus on other stuff. I will try to have a look shortly.

[gpastal24]

Hi @keniel123 @oscarfossey . Sorry for the long wait, I just found some time today to do that.
I have added it to a different branch. I haven't tested TRT support, but the visualization is OK. I have only included the VitposeB+ config. The code in general is spaghetti code, and I dont have time to work on this :)

[oscarfossey]

Thanks for the reply, wish you luck on your projects