Learning-Based Post-Processing methods

May 20, 2022 ยท View on GitHub

We provide two learning-based post-processing methods deciwatch and smoothnet for speed-up and smoothing respectively.

DeciWatch

We provide the config files for DeciWatch: DeciWatch: A Simple Baseline for 10x Efficient 2D and 3D Pose Estimation.

@article{zeng2022deciwatch,
  title={DeciWatch: A Simple Baseline for 10x Efficient 2D and 3D Pose Estimation},
  author={Zeng, Ailing and Ju, Xuan and Yang, Lei and Gao, Ruiyuan and Zhu, Xizhou and Dai, Bo and Xu, Qiang},
  journal={arXiv preprint arXiv:2203.08713},
  year={2022}
}

Notes

We use checkpoints trained on SPIN-3DPW for demo speed up. Checkpoints with different intervals and q values are provided. If you need more checkpoints trained on various datasets and backbones, please refer to the official implementation of DeciWatch.

IntervalWindow QConfigDownloadSpeed UpPrecision Improvement (MPJPE In/Out)
101deciwatch_interval10_q1model10X99.35 / 95.85
102deciwatch_interval10_q2model10X99.45 / 96.37
103deciwatch_interval10_q3model10X99.60 / 96.98
104deciwatch_interval10_q4model10X99.58 / 96.87
105deciwatch_interval10_q5model10X99.78 / 97.39
51deciwatch_interval5_q1model5X99.31 / 95.05
52deciwatch_interval5_q2model5X99.35 / 95.05
53deciwatch_interval5_q3model5X99.45 / 94.84
54deciwatch_interval5_q4model5X99.45 / 94.94
55deciwatch_interval5_q5model5X99.55 / 94.48

To use different settings of DeciWatch in demo, specify --speed_up_type with the checkpoint name. For example, you may use --speed_up_type deciwatch_interval10_q3 for 10X speed up with a window size of 31. Simply set --speed_up_type deciwatch to use default setting deciwatch_interval5_q3. The meaning of interval and q can be found in the original paper.

SmoothNet

We provide the config files for SmoothNet: A Plug-and-Play Network for Refining Human Poses in Videos.

@article{zeng2021smoothnet,
  title={SmoothNet: A Plug-and-Play Network for Refining Human Poses in Videos},
  author={Zeng, Ailing and Yang, Lei and Ju, Xuan and Li, Jiefeng and Wang, Jianyi and Xu, Qiang},
  journal={arXiv preprint arXiv:2112.13715},
  year={2021}
}

Notes

We use checkpoints trained on SPIN-3DPW for demo pose smoothing. Checkpoints with different window size are provided.

Window SizeConfigDownloadPrecision Improvement (MPJPE In/Out)Smoothness Improvement (Accel In/Out)
8smoothnet_windowsize8model96.85 / 95.8434.62 / 7.13
16smoothnet_windowsize16model96.85 / 95.6134.62 / 6.35
32smoothnet_windowsize32model96.85 / 95.0334.62 / 6.11
64smoothnet_windowsize64model96.85 / 95.2634.62 / 6.02

To use different settings of SmoothNet in demo, specify --smooth_type with the checkpoint name. For example, you may use --smooth_type smoothnet_windowsize8 for pose smoothing with a window size of 8. Simply set --mooth_type smoothnet to use default setting smoothnet_windowsize8. The meaning of windowsize can be found in the original paper.