InfantNet: A Large Scale Dataset for Infant Body Pose and Shape Estimation
Organizations: Johns Hopkins University · Sony Corporation of America · University of Washington · CISPA Helmholtz Center for Information Security
Abstract
Infant pose and shape estimation is essential for applications in childcare, developmental monitoring, and medical diagnosis. However, existing methods and datasets are largely designed for adults, and direct transfer to infants fails due to substantial differences in body proportions, articulation limits, and frequent self-occlusion. To address this gap, we introduce InfantNet, the largest real-image infant dataset to date, comprising 108,902 RGB images of infants aged 6-18 months. Each image is annotated with 2D keypoints, and a curated subset of 11,642 images additionally includes 3D pose and shape annotations with full SMIL parameters, verified through multi-view rendering and reprojection-error filtering. We use an iterative annotation pipeline to ensure high fidelity across both 2D and 3D labels. InfantNet establishes a large-scale, comprehensive benchmark for infant 2D keypoint detection and 3D pose-and-shape recovery. Baseline experiments demonstrate that state-of-the-art adult pose estimators do not generalize well to infants, whereas fine-tuning on InfantNet yields a consistent improvement. The gains are even more pronounced for 3D pose and shape estimation, where SMIL-based estimation demonstrates robust performance with accuracy comparable to results reported for SMPL-based models. By releasing the InfantNet dataset and benchmark, we provide a vital resource for advancing infant pose analysis and related healthcare applications.
Figures & tables
| Dataset | Syn/Real | # Images | Keypoints | Segmentation | 3D Annotations |
|---|---|---|---|---|---|
| BabyPose [ 35 ] | Real | 16,000 | Yes | No | Depth only |
| BabyNet [ 12 ] | Real | 2,984 | No | BBox only | No |
| MINI-RGBD [ 17 ] | Synthetic | 12,000 | Yes | No | Depth only |
| SyRIP [ 19 ] | Syn+Real | 1,700 | Yes | No | No |
| InfantNet (ours) | Real | 108,902 | Yes | Yes | Yes |
| Subset | Viewpoint Info | Raw Frames | Filtered Frames |
|---|---|---|---|
| YouTube (Train) | Varied | 44,257 | 36,185 |
| YouTube (Test) | Varied | 6,782 | 3,959 |
| HomeVisit (Train) | 4 fixed cameras | 123,876 | 56,968 |
| HomeVisit (Test) | 4 fixed cameras | 31,521 | 11,790 |
| Total | – | 206,436 | 108,902 |
| Model | Direct Inference (%) | Fine-tuned (%) | ||||||
|---|---|---|---|---|---|---|---|---|
| AP | AP.5 | AP.75 | AR | AP | AP.5 | AP.75 | AR | |
| Hourglass 256x256 | 86.27 | 95.58 | 92.37 | 88.43 | 92.63 | 98.99 | 97.90 | 93.95 |
| Hourglass 384x384 | 87.39 | 96.59 | 93.43 | 89.27 | 93.40 | 98.98 | 97.89 | 94.52 |
| HRFormer-small 256x192 | 87.65 | 96.53 | 93.36 | 89.82 | 93.14 | 98.97 | 97.87 | 94.36 |
| HRFormer-small 384x288 | 88.00 | 96.54 | 94.30 | 90.07 | 93.51 | 98.97 | 97.91 | 94.60 |
| HRFormer-base 256x192 | 88.24 | 95.50 | 93.33 | 90.26 | 93.85 | 98.98 | 97.92 | 95.09 |
| Model | Adult Body Model | Infant Body Model | ||
|---|---|---|---|---|
| MPJPE | PA-MPJPE | MPJPE | PA-MPJPE | |
| HMR [ 22 ] | 393.30 | 124.15 | 103.36 ( 74% ) | 58.79 ( 53% ) |
| PARE [ 25 ] | 364.49 | 120.06 | 83.73 ( 77% ) | 63.16 ( 47% ) |
| 4D-Humans [ 15 ] | 526.62 | 125.02 | 84.73 ( 84% ) | 46.74 ( 63% ) |
| Method | All | Head | Torso | Arms | Legs |
|---|---|---|---|---|---|
| ViTPose-H (Before FT) | 0.871 | 0.962 | 0.842 | 0.865 | 0.793 |
| ViTPose-H (FT on InfantNet) | 0.906 | 0.986 | 0.890 | 0.901 | 0.828 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Direct Inference | Source-Specific Fine-tuned | Cross-Subset Fine-tuned | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| AP | AP.5 | AP.75 | AR | AP | AP.5 | AP.75 | AR | AP | AP.5 | AP.75 | AR | |
| Hourglass 256x256 | 0.8588 | 0.9790 | 0.9440 | 0.8868 | 0.9193 | 0.9898 | 0.9678 | 0.9374 | 0.8667 | 0.9896 | 0.9566 | 0.8962 |
| Hourglass 384x384 | 0.8707 | 0.9794 | 0.9452 | 0.8987 | 0.9284 | 0.9897 | 0.9781 | 0.9450 | 0.8808 | 0.9894 | 0.9566 | 0.9091 |
| HRFormer-small_256x192 | 0.8761 | 0.9779 | 0.9537 | 0.9062 | 0.9247 | 0.9896 | 0.9778 | 0.9422 | 0.8734 | 0.9896 | 0.9549 | 0.9023 |
| HRFormer-small_384x288 | 0.8852 | 0.9887 | 0.9557 | 0.9145 | 0.9307 | 0.9895 | 0.9788 | 0.9475 | 0.8816 | 0.9896 | 0.9658 | 0.9115 |
| HRFormer-base_256x192 | 0.8900 | 0.9883 | 0.9559 | 0.9172 | 0.9325 | 0.9893 | 0.9782 | 0.9489 | 0.8964 | 0.9898 | 0.9673 | 0.9235 |
| Model | Direct Inference | Source-Specific Fine-tuned | Cross-Subset Fine-tuned | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| AP | AP.5 | AP.75 | AR | AP | AP.5 | AP.75 | AR | AP | AP.5 | AP.75 | AR | |
| Hourglass 256x256 | 0.8643 | 0.9570 | 0.9257 | 0.8834 | 0.9222 | 0.9898 | 0.9689 | 0.9357 | 0.9116 | 0.9799 | 0.9687 | 0.9273 |
| Hourglass 384x384 | 0.8724 | 0.9572 | 0.9358 | 0.8907 | 0.9288 | 0.9897 | 0.9790 | 0.9415 | 0.9177 | 0.9797 | 0.9689 | 0.9329 |
| HRFormer-small_256x192 | 0.8766 | 0.9573 | 0.9359 | 0.8954 | 0.9235 | 0.9896 | 0.9687 | 0.9369 | 0.9194 | 0.9796 | 0.9685 | 0.9350 |
| HRFormer-small_384x288 | 0.8774 | 0.9568 | 0.9352 | 0.8960 | 0.9261 | 0.9893 | 0.9688 | 0.9398 | 0.9219 | 0.9896 | 0.9688 | 0.9369 |
| HRFormer-base_256x192 | 0.8769 | 0.9560 | 0.9346 | 0.8977 | 0.9326 | 0.9897 | 0.9791 | 0.9452 | 0.9260 | 0.9893 | 0.9688 | 0.9423 |
| Model | MPJPE | PA-MPJPE |
|---|---|---|
| HMR [ 22 ] | 118.53 | 62.30 |
| PARE [ 25 ] | 107.52 | 63.83 |
| 4D-Humans [ 15 ] | 84.50 | 42.40 |
| Model | Avg Time (s / img) |
|---|---|
| Optimization | |
| SMIL-Opt [ 3 , 45 ] | 38.7020 |
| End-to-End Models | |
| HMR [ 22 ] | 0.0033 |
| PARE [ 25 ] | 0.0091 |
| 4D-Humans [ 15 ] | 0.0212 |
| Model | MPJPE | PA-MPJPE |
|---|---|---|
| HMR [ 22 ] | 100.00 | 55.60 |
| PARE [ 25 ] | 78.48 | 59.41 |
| 4D-Humans [ 15 ] | 86.57 | 50.15 |
Human Pose Estimation
Human pose estimation recovers people’s body keypoints, 3D skeletons, or meshes from images, video, and other sensor data.
See all 80 papers