Algorithms

August 17, 2022 · View on GitHub

Introduction

We provide the config files for training and pretrained models for inference on the optimal configurations.

Notes

Download the pretrained backbones from here:

  1. resnet50_coco_pose.pth
  2. hrnet_coco_pose.pth
  3. hrnetw48_coco_pose.pth

Download the above resources and arrange them in the following file structure:

mmhuman3d
├── mmhuman3d
├── docs
├── tests
├── tools
├── configs
└── data
    └── checkpoints
        ├── resnet50_coco_pose.pth
        ├── hrnet_coco_pose.pth
        └── hrnetw48_coco_pose.pth

Results and Models

We evaluate on 3DPW. Values are PA-MPJPE.

Our baseline models for HMR, SPIN and GraphCMR can reach the reported results in the respective works. For PARE, the original work trains the model on MPII for pose estimation task and later on EFT-COCO for mesh recovery before training on he full set of datasets. To keep consistent with the practice adopted throughout our work, we benchmark PARE by training it fro scratch with only ImageNet initialisation. For Graphormer, the original work evaluates on H36M every epoch before fine-tuning the best H36M model on 3DPW-train (Protocol 1) for 5 epochs. To keep consistent, we adopt Protocol 2 instead.

This is a summary of our results:

AlgorithmsDatasetsBackboneInitialisationNormalL1L1+COCOL1+COCO+Aug
HMRH36M, MI, COCO, LSP, LSPET, MPIIResNet-50ImageNet64.5558.2051.8051.66
SPINH36M, MI, COCO, LSP, LSPET, MPIIResNet-50HMR(ImageNet)59.0057.0851.5450.69
GraphCMRH36M, COCO, LSP, LSPET, MPII, UP3DResNet-50ImageNet70.5167.2061.7460.26
PAREH36M, MI, EFT-[COCO, LSPET, MPII]HRNet-W32ImageNet61.9961.1359.9858.32
GraphormerH36M, MuCo, COCO, UP3D, MPIIHRNet-W48ImageNet63.1863.4759.6658.82

These are the provided logs:

ConfigDatasetsBackboneVariant3DPWLog
HMRH36M, MI, COCO, LSP, LSPET, MPIIResNet-50Normal64.55log
HMRH36M, MI, COCO, LSP, LSPET, MPIIResNet-50L158.20log
HMRH36M, MI, COCO, LSP, LSPET, MPIIResNet-50L1 + COCO51.8log
HMRH36M, MI, COCO, LSP, LSPET, MPIIResNet-50L1 + COCO + Aug51.66log
SPINH36M, MI, COCO, LSP, LSPET, MPIIResNet-50Normal59.00log
SPINH36M, MI, COCO, LSP, LSPET, MPIIResNet-50L157.08log
SPINH36M, MI, COCO, LSP, LSPET, MPIIResNet-50L1 + COCO51.54log
SPINH36M, MI, COCO, LSP, LSPET, MPIIResNet-50L1 + COCO + Aug50.69log
GraphCMRH36M, COCO, LSP, LSPET, MPII, UP3DResNet-50Normal70.51log
GraphCMRH36M, COCO, LSP, LSPET, MPII, UP3DResNet-50L167.2log
GraphCMRH36M, COCO, LSP, LSPET, MPII, UP3DResNet-50L1 + COCO61.74log
GraphCMRH36M, COCO, LSP, LSPET, MPII, UP3DResNet-50L1 + COCO + Aug60.26log
PAREH36M, MI, EFT-[COCO, LSPET, MPII]HRNet-W32Normal61.99log
PAREH36M, MI, EFT-[COCO, LSPET, MPII]HRNet-W32L161.13log
PAREH36M, MI, EFT-[COCO, LSPET, MPII]HRNet-W32L1 + COCO59.98log
PAREH36M, MI, EFT-[COCO, LSPET, MPII]HRNet-W32L1 + COCO + Aug58.32log
GraphormerH36M, MuCo, COCO, UP3D, MPIIHRNet-W48Normal63.18log
GraphormerH36M, MuCo, COCO, UP3D, MPIIHRNet-W48L163.47log
GraphormerH36M, MuCo, COCO, UP3D, MPIIHRNet-W48L1 + COCO59.66log
GraphormerH36M, MuCo, COCO, UP3D, MPIIHRNet-W48L1 + COCO + Aug58.82log