Skip to content

Latest commit

 

History

History
51 lines (28 loc) · 1.19 KB

pose3d.md

File metadata and controls

51 lines (28 loc) · 1.19 KB
Raw

3D Human Pose Estimation

Data

  1. Download the finetuned Stacked Hourglass detections and our preprocessed H3.6M data here and unzip it to data/motion3d.

Note that the preprocessed data is only intended for reproducing our results more easily. If you want to use the dataset, please register to the Human3.6m website and download the dataset in its original format. Please refer to LCN for how we prepare the H3.6M data.

  1. Slice the motion clips (len=243, stride=81)

    python tools/convert_h36m.py

Running

Train from scratch:

python train.py \
--config configs/pose3d/MB_train_h36m.yaml \
--checkpoint checkpoint/pose3d/MB_train_h36m

Finetune from pretrained MotionBERT:

python train.py \
--config configs/pose3d/MB_ft_h36m.yaml \
--pretrained checkpoint/pretrain/MB_release \
--checkpoint checkpoint/pose3d/FT_MB_release_MB_ft_h36m

Evaluate:

python train.py \
--config configs/pose3d/MB_train_h36m.yaml \
--evaluate checkpoint/pose3d/MB_train_h36m/best_epoch.bin