This repository contains the official implementation of our paper: Human Mesh Recovery from Arbitrary Multi-view Images. Xiaoben Li, Mancheng Meng, Ziyan Wu, Terrence Chen, Fan Yang*, Dinggang Shen. paper
Abstract: Human mesh recovery from arbitrary multi-view images involves two characteristics: the arbitrary camera poses and arbitrary number of camera views. Because of the variability, designing a unified framework to tackle this task is challenging. The challenges can be summarized as the dilemma of being able to simultaneously estimate arbitrary camera poses and recover human mesh from arbitrary multi-view images while maintaining flexibility. To solve this dilemma, we propose a divide and conquer framework for Unified Human Mesh Recovery (U-HMR) from arbitrary multi-view images. In particular, U-HMR consists of a decoupled structure and two main components: camera and body decoupling (CBD), camera pose estimation (CPE), and arbitrary view fusion (AVF). As camera poses and human body mesh are independent of each other, CBD splits the estimation of them into two sub-tasks for two individual sub-networks (\ie, CPE and AVF) to handle respectively, thus the two sub-tasks are disentangled. In CPE, since each camera pose is unrelated to the others, we adopt a shared MLP to process all views in a parallel way. In AVF, in order to fuse multi-view information and make the fusion operation independent of the number of views, we introduce a transformer decoder with a SMPL parameters query token to extract cross-view features for mesh recovery. To demonstrate the efficacy and flexibility of the proposed framework and effect of each component, we conduct extensive experiments on three public datasets: Human3.6M, MPI-INF-3DHP, and TotalCapture.
- OS: Linux
- python: 3.9.12
- pytorch: 1.11.0
You can create a conda environment use the following commands.
conda create --name U-HMR
source activate U-HMR
pip install -r requirements.txt
- Human3.6M: We follow the H36M-Toolbox to process Human3.6M dataset. And the SMPL annotations are from PyMAF.
- MPI-INF-3DHP: We follow the process code from the dataset and the code from SPIN.
- TotalCapture: We follow the TotalCature-Toolbox and the code from the dataset.
- Others: Most of other data are from SPIN and 4D-Humans.
We provide a set of pretrained models (password:uhmr) as follows:
- ResNet50
- MLP
- Independent tokens
- Decoupling with MLP
- Decoupling with Transformer decoder
The general structure of the project is as follows.
- experiments: config files that contain arguments for performing experiments
- lib: core code, contains datasets, models and utils
- logs: tensorboard log directory
- output: training model output directory
- run.py: training and testing code that call the dataset and model functions
python run.py --cfg_name cfg_file.yaml --dataset dataset_name
- Tested on Human3.6M
MPJPE | PA-MPJPE | |
---|---|---|
Ours-SV(ResNet50) | 52.9 | 42.5 |
Ours-SV(ViT) | 43.3 | 32.6 |
Ours-MV(ResNet50) | 36.3 | 28.3 |
Ours-MV(ViT) | 31.0 | 22.8 |
This work could not have been done without contributions from other open source projects, and we sincerely thank the contributor of those projects.
If you find this repository useful, please kindly cite our paper.
@misc{li2024uhmr,
title={Human Mesh Recovery from Arbitrary Multi-view Images},
author={Xiaoben Li and Mancheng Meng and Ziyan Wu and Terrence Chen and Fan Yang and Dinggang Shen},
year={2024},
eprint={2403.12434},
archivePrefix={arXiv}
}