TFAN

Twin-Flow Axis Normalization for Human Motion Prediction

Network Architecture

---

Requirements

---

• PyTorch == 1.12
• Numpy
• CUDA == 11.6
• Easydict
• pickle
• einops
• scipy
• six

Data Preparation

---

Download all the data and put them in the ./data directory.

H3.6M

AMASS download the list (ACCAD, MPI_Limits, CMU, Eyes_Japan_Dataset, KIT, EKUT, TotalCapture, TCD_handMocap, BioMotionLab_NTroje) and we need SMPL+H G.

3DPW

CMU-Mocap

Training

---

H3.6M

position

cd exp/h36m/position/
python train.py

bone

cd exp/h36m/bone/
python train.py

fusion

cd exp/h36m/fusion/
python train.py

AMASS

position

cd exp/amass_3dpw/position/
python train.py

bone

cd exp/amass_3dpw/bone/
python train.py

fusion

cd exp/amass_3dpw/fusion/
python train.py

####CMU-Mocap

position

cd exp/cmu/position/
python train.py

bone

cd exp/cmu/bone/
python train.py

fusion

cd exp/cmu/fusion/
python train.py

Evaluation

---

H3.6M

cd exps/h36m/fusion/
python test.py

AMASS_3DPW

cd exps/amass_3dpw/fusion/
python test.py

CMU-Mocap

cd exps/cmu/fusion/
python test.py