Update README.md
Browse files
README.md
CHANGED
|
@@ -7,4 +7,25 @@ tags:
|
|
| 7 |
- 2D_Pose_Estimation
|
| 8 |
- MMPOSE
|
| 9 |
- RTMO
|
| 10 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
- 2D_Pose_Estimation
|
| 8 |
- MMPOSE
|
| 9 |
- RTMO
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# Retrainable RTMO-s Model
|
| 13 |
+
|
| 14 |
+
This repository provides a fully **retrainable** RTMO-s checkpoint for 2D human pose estimation in the MMPOSE framework.
|
| 15 |
+
|
| 16 |
+
The RTMO one-stage model family (variants T, S, M, L) was originally released by the OpenMMLab team with pre-trained weights, but those official PyTorch `.pth` checkpoints do **not** preserve all parameter keys—making fine-tuning or re-training impossible within MMPOSE (see [discussion #3076](https://github.com/open-mmlab/mmpose/discussions/3076)).
|
| 17 |
+
|
| 18 |
+
To address this limitation, PESI has faithfully reproduced the RTMO-s training procedure on the MS COCO 2017 dataset using the exact configuration from the official MMPOSE RTMO project. Our checkpoint preserves every model key, enabling you to fine-tune on custom datasets (e.g. Body7, MPII) or continue training from this strong baseline.
|
| 19 |
+
|
| 20 |
+
- **Original implementation & configs**: [MMPOSE RTMO project](https://github.com/open-mmlab/mmpose/tree/main/projects/rtmo)
|
| 21 |
+
- **Issue on missing keys**: [MMPOSE #3076](https://github.com/open-mmlab/mmpose/discussions/3076)
|
| 22 |
+
|
| 23 |
+
## Key Features
|
| 24 |
+
|
| 25 |
+
- **One-stage RTMO-s architecture** as described in the MMPOSE RTMO project
|
| 26 |
+
- **All keys retained** in the PyTorch checkpoint for full retrainability (unlike the official weights)
|
| 27 |
+
|
| 28 |
+
## References
|
| 29 |
+
|
| 30 |
+
1. OpenMMLab MMPOSE RTMO project: “RTMO: Real-time One-Stage Multi-person Pose Estimation”
|
| 31 |
+
2. Issue discussion on missing keys in official RTMO checkpoints
|