IDEA-Research / DWPose

"Effective Whole-body Pose Estimation with Two-stages Distillation" (ICCV 2023, CV4Metaverse Workshop)
Apache License 2.0
2.19k stars 141 forks source link

Python script to run this? #50

Closed landmann closed 1 year ago

landmann commented 1 year ago

Anyone have a python script I can use to detect the whole body pose of objects?

Not sure what I should do to just get a pose of an image 😕 Should I do pip install this file and then get the annotator? but I'd have to download the weights? or if I download the hf_hub files, there are so many and I don't know what any of them mean lol

Would be greatly appreciated!

yzd-v commented 1 year ago

This topdown file works with .pth model. This pose file works with .onnx model. Both of them are fine to get an image.