Add fine-tunable rtmo-s tuned on body6
Browse files
app.py
CHANGED
|
@@ -49,6 +49,7 @@ REMOTE_CHECKPOINTS = {
|
|
| 49 |
"rtmo-l_16xb16-700e_crowdpose": "https://download.openmmlab.com/mmpose/v1/projects/rtmo/rtmo-l_16xb16-700e_crowdpose-640x640-1008211f_20231211.pth",
|
| 50 |
# Retrainable from HF repo
|
| 51 |
"rtmo-s_coco_retrainable": "https://huggingface.co/Luigi/Retrainable-RTMO-s/resolve/main/rtmo-s_coco_retrainable.pth",
|
|
|
|
| 52 |
}
|
| 53 |
|
| 54 |
# Variants for inference (prefixes)
|
|
|
|
| 49 |
"rtmo-l_16xb16-700e_crowdpose": "https://download.openmmlab.com/mmpose/v1/projects/rtmo/rtmo-l_16xb16-700e_crowdpose-640x640-1008211f_20231211.pth",
|
| 50 |
# Retrainable from HF repo
|
| 51 |
"rtmo-s_coco_retrainable": "https://huggingface.co/Luigi/Retrainable-RTMO-s/resolve/main/rtmo-s_coco_retrainable.pth",
|
| 52 |
+
"rtmo-s_body6_retrainable": "https://huggingface.co/Luigi/Retrainable-RTMO-s/resolve/main/body6_epoch_600.pth",
|
| 53 |
}
|
| 54 |
|
| 55 |
# Variants for inference (prefixes)
|