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
|