search

bmartacho / UniPose

We propose UniPose, a unified framework for human pose estimation, based on our “Waterfall” Atrous Spatial Pooling architecture, that achieves state-of-art-results on several pose estimation metrics. Current pose estimation methods utilizing standard CNN architectures heavily rely on statistical postprocessing or predefined anchor poses for joint localization. UniPose incorporates contextual seg- mentation and joint localization to estimate the human pose in a single stage, with high accuracy, without relying on statistical postprocessing methods. The Waterfall module in UniPose leverages the efficiency of progressive filter- ing in the cascade architecture, while maintaining multi- scale fields-of-view comparable to spatial pyramid config- urations. Additionally, our method is extended to UniPose- LSTM for multi-frame processing and achieves state-of-the- art results for temporal pose estimation in Video. Our re- sults on multiple datasets demonstrate that UniPose, with a ResNet backbone and Waterfall module, is a robust and efficient architecture for pose estimation obtaining state-of- the-art results in single person pose detection for both sin- gle images and videos.
Other
211 stars 44 forks source link

Inference or demo is missing #21

Closed simsim314 closed 3 years ago

simsim314 commented 3 years ago

Hi I understand it's a new work - and you published your pretrained model which is great.

But as an end user who wants to check out your model quality for his purposes there is no simple code snippet to run inference, and no demo. It looks you have already written the test function and also have all the components for this.

Yet no readme nor some basic indication how to infer the model on an image.

bmartacho commented 3 years ago

Thank you for the interest in UniPose. Please refer to the following issue addressing your question.

https://github.com/bmartacho/UniPose/issues/8#issuecomment-791405250

A-qingtongxiaoyou commented 1 year ago

Hi, I found the Inference by following #8 . But the Inference for a video still confuse me. https://github.com/bmartacho/UniPose/blob/master/uniposeLSTM.py Line 223

    def test(self,epoch):
        self.model.eval()
        print("Testing") 

        img_path = '/PATH/TO/TEST/IMAGE'

It seems still the interface of a image. @bmartacho