LibreRFDETRm-pose / README.md
Xuban's picture
Update experimental RF-DETR pose checkpoint
85e6125 verified
|
Raw
History Blame Contribute Delete
2.34 kB
---
license: apache-2.0
library_name: libreyolo
tags:
- computer-vision
- pose-estimation
- keypoint-detection
- coco
- libreyolo
- rfdetr
datasets:
- detection-datasets/coco
---
# LibreRFDETRm-pose
**EXTREMELY experimental** RF-DETR-m pose checkpoint for LibreYOLO.
This is a COCO-17 human pose preview checkpoint for LibreYOLO's `task="pose"` RF-DETR path. It is useful for testing and bootstrapping, but it is not a final benchmark release.
## Checkpoint
- File: `LibreRFDETRm-pose.pt`
- Family: `LibreRFDETR`
- Size: `m`
- Task: `pose`
- Classes: person only
- Keypoints: COCO-17, `(x, y, visibility)`
- Validation image size: `576`
- Additional training epochs for this checkpoint: `0`
## Initialization Method
Native RF-DETR-m detection checkpoint plus shared tensors from the trained LibreRFDETRs-pose checkpoint. The extra final decoder layer was initialized from the trained small-pose final decoder layer.
This method keeps the size-specific detection backbone and resolution-dependent tensors, then transfers the pose-specialized shared tensors from the small pose checkpoint. The checkpoint should still be treated as experimental until a full per-size training run is published.
## COCO Keypoint Validation
Validation was run on COCO person keypoints val2017 through LibreYOLO's pose validator.
| Metric | Value |
| --- | ---: |
| keypoints mAP50-95 | `0.532909` |
| keypoints mAP50 | `0.837690` |
| keypoints mAP75 | `0.581342` |
| keypoints AR50-95 | `0.641814` |
The validation artifacts are included as `validation_metrics.json`. Initialization details are included as `initialization_summary.json`.
## Usage
```python
from libreyolo import LibreRFDETR
model = LibreRFDETR("LibreRFDETRm-pose.pt", task="pose")
results = model.predict("image.jpg", imgsz=576)
print(results[0].keypoints)
```
Autodownload in LibreYOLO emits an experimental warning for this checkpoint.
## Caveats
- Experimental checkpoint, not a final benchmark release.
- No additional fine-tuning epochs were run for this per-size checkpoint after transfer initialization.
- Pose export/runtime backends may have separate support status from PyTorch inference.
- Metrics are from LibreYOLO PR development artifacts, not from an independent external benchmark suite.