redrock303 / HEMlets

17 stars 2 forks source link

HEMlets Pose: Learning Part-Centric Heatmap Triplets for Accurate 3D Human Pose Estimation

Kun Zhou, Xiaoguang Han, Nianjuan Jiang, Kui Jia, Jiangbo Lu

Here we provide our implementation of HEMlets PoSh: Learning Part-Centric Heatmap Triplets for 3D Human Pose and Shape Estimation. Please follow the links to read our ICCV'19, TPAMI'21 and visit the corresponding project page.

We provide quick inference code to validate and visualize our results on Human3.6M. Brief runing instructions are given below.

  1. Pre installation\ create a new conda vitual environment\ conda/pip install -r requirement.txt
  2. Download the pre-trained model and the tiny dataset(a pre-processed testing video from Human3.6M) at Baidu Cloud [code:HEMs] or Google Cloud.
  3. Visualization and evaluation on a single video from Human3.6M. We implement a script for visualization and evaluation of the predicted results on Human3.6M by running the command:\ bash inference.sh\ if 'visualize' is set to 1, it means the visualization is activated, and you will get an additional video file in the root path which records all the rendering frames. Otherwise, it will only print the P1/P2 result on the screen.\ \ We will also plan to update this repository with the training code later.


### Citing
If you find this code useful for your research, please consider citing the following paper:
   @Inproceedings{zhou2019hemlets,
     Title          = {HEMlets Pose: Learning Part-Centric Heatmap Triplets for Accurate 3D Human Pose Estimation},
     Author         = {Kun Zhou, Xiaoguang Han, Nianjuan Jiang, Kui Jia, Jiangbo Lu},
     Booktitle      = {International Conference on Computer Vision (ICCV)},
     Year           = {2019}
   }

   @article{zhou2020hemlets,
     title           = {HEMlets PoSh: Learning Part-Centric Heatmap Triplets for 3D Human Pose and Shape  Estimation},
     Author          = {Kun Zhou, Xiaoguang Han, Nianjuan Jiang, Kui Jia, Jiangbo Lu},
     Journal         = {IEEE Trans. on Pattern Analysis and Machine Intelligence (TPAMI)},
     Year            = {2021}
    }

### Notices
> This code implementation is only for research or educational purposes, and not freely available for commercial use or redistribution. 
If you have any problems or suggestions, please feel free to contact Kun Zhou at 1039557638@qq.com.