Comparison of marker-less 2D image-based methods for infant pose estimation

TL;DR

This study compares generic and infant-specific 2D pose estimation methods for infants, finding that generic models perform well and that top-down camera views improve accuracy, informing better recording practices for infant movement analysis.

Contribution

It demonstrates that generic adult-trained pose estimators perform effectively on infants and highlights the benefits of top-down views for improved pose estimation accuracy.

Findings

ViTPose performs best among generic models on infants.

Retraining generic models improves infant pose estimation accuracy.

Top-down view significantly enhances pose detection, especially for hips.

Abstract

In this study we compare the performance of available generic- and infant-pose estimators for a video-based automated general movement assessment (GMA), and the choice of viewing angle for optimal recordings, i.e., conventional diagonal view used in GMA vs. top-down view. We used 4500 annotated video-frames from 75 recordings of infant spontaneous motor functions from 4 to 26 weeks. To determine which pose estimation method and camera angle yield the best pose estimation accuracy on infants in a GMA related setting, the distance to human annotations and the percentage of correct key-points (PCK) were computed and compared. The results show that the best performing generic model trained on adults, ViTPose, also performs best on infants. We see no improvement from using infant-pose estimators over the generic pose estimators on our infant dataset. However, when retraining a generic model on our data, there is a significant improvement in pose estimation accuracy. The pose estimation accuracy obtained from the top-down view is significantly better than that obtained from the diagonal view, especially for the detection of the hip key-points. The results also indicate limited generalization capabilities of infant-pose estimators to other infant datasets, which hints that one should be careful when choosing infant pose estimators and using them on infant datasets which they were not trained on. While the standard GMA method uses a diagonal view for assessment, pose estimation accuracy significantly improves using a top-down view. This suggests that a top-down view should be included in recording setups for automated GMA research.