Skip to content

pytorch version of "End-to-end Recovery of Human Shape and Pose"

Notifications You must be signed in to change notification settings

MandyMo/pytorch_HMR

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

54 Commits
 
 
 
 
 
 

Repository files navigation

HMR

This is a pytorch implementation of End-to-end Recovery of Human Shape and Pose by Angjoo Kanazawa, Michael J. Black, David W. Jacobs, and Jitendra Malik, accompanying by some famous human pose estimation networks and datasets. HMR is an end-to end framework for reconstructing a full 3D mesh of a human body from a single RGB image. In contrast to most current methods that compute 2D or 3D joint locations, HMR produce a richer and more useful mesh representation that is parameterized by shape and 3D joint angles. The main objective is to minimize the reprojection loss of keypoints, which allow model to be trained using in-the-wild images that only have ground truth 2D annotations. For visual impact, please visit the author's original video.

training step (the following links are not available now due to license limitation)

1. download the following datasets.

2. download human3.6 datasets.

3. unzip the downloaded datasets.

4. unzip the model.zip

5. config the environment by modify the src/config.py and do_train.sh

6. run ./do_train.sh directly

environment configurations.

  • install pytorch0.4
  • install torchvision
  • install numpy
  • install scipy
  • install h5py
  • install opencv-python

result

reference papers

reference resources