Luigi commited on
Commit
e2f80db
·
verified ·
1 Parent(s): 486c196

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -1
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