• Semiconductor Optoelectronics
  • Vol. 43, Issue 5, 968 (2022)
LI Jie1,2,3, QI Bo1,2,3, and ZHANG Jianlin2,3
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • 3[in Chinese]
  • show less
    DOI: 10.16818/j.issn1001-5868.2022022301 Cite this Article
    LI Jie, QI Bo, ZHANG Jianlin. A Testing-Time-Augmentation Algorithm for Single Human Pose Estimation Based on Aleatoric Uncertainty[J]. Semiconductor Optoelectronics, 2022, 43(5): 968 Copy Citation Text show less

    Abstract

    Aiming at the problem of lacking reliability evaluation and robustness guarantee in existing single-person pose estimation networks′ results, a testing-time-augmentation (TTA) algorithm based on aleatoric uncertainty was proposed. In this TTA algorithm, diverse outputs were firstly obtained by stochastic parallel data augmentation and model inference. Then, the reliability evaluations of those outputs are acquired by calculating their aleatoric uncertainty. Finally, those outputs and their uncertainty were fused according to the reliabilities to obtain a more accurate and robust result as well as its evaluation. Experiments on the MPII dataset show that this algorithm can be applied to any existing single-person pose estimation network in a plug-and-play manner, leading to a more precise and robust result with its uncertainty evaluation.