Our PointHMR can be run by following command
We conduct large-scale training on multiple 2D and 3D datasets, including Human3.6M, COCO, MUCO, UP3D, MPII. During training, it will evaluate the performance per epoch, and save the best checkpoints.
sh script/pretrain.sh
-
nproc_per_node: Number of GPU.
-
train_yaml: Path of training datasets (Human3.6M, COCO, MUCO, UP3D, MPII).
-
val_yaml: Path of valiadation dataset (Human3.6M).
We follow prior works that also use 3DPW training data. During training, it will evaluate the performance per epoch, and save the best checkpoints. We fine-tune on 3DPW training set with the checkpoint from the pretraining scheme by the following command.
sh script/finetuning.sh
-
nproc_per_node: Number of GPU.
-
train_yaml: Path of training dataset (3DPW).
-
val_yaml: Path of validation dataset (3DPW).
-
resume_checkpoint: Path to the specific checkpoint (checkpoint from pretraining) for resume training.
We evaluate our PointHMR on the Human3.6M dataset by the following command.
sh script/evaluation_h36m.sh
-
nproc_per_node: Number of GPU.
-
val_yaml: Path of validation dataset (Human3.6M).
-
resume_checkpoint: Path to the specific checkpoint for evaluation.
We evaluate our PointHMR on the 3DPW dataset by the following command.
sh script/evalutation_3dpw.sh
- nproc_per_node: Number of GPU.
- val_yaml: Path of validation dataset (3DPW).
- resume_checkpoint: Path to the specific checkpoint for evaluation.