search

bmartacho / OmniPose

This is the official PyTorch implementation for “OmniPose: A Multi-Scale Framework for Multi-Person Pose Estimation”.
Other
64 stars 8 forks source link

Evaluation and inference with pretrained models #7

Open mthodoris opened 1 year ago

mthodoris commented 1 year ago

Trying to run the test.py and inference.py using pretrained_models returns 0.0 avg_precision and random keypoints inside the images. Are the provided models pre-trained? Is there anything that I am missing?

000000301563 000000177861

mircodemarchi commented 1 year ago

I'm encountering the same problem

liukp557 commented 1 year ago

I had the same problem, have you solved it yet?

bblgbr commented 7 months ago

I solved it and I found that the author provided the WASPv1 pretrained model. Therefore, you need to change the OmniPose/models/omnipose.py to waspv1