Skip to content

feat: add RF-DETR keypoint preview model#1257

Merged
chmjkb merged 6 commits into
mainfrom
@chmjkb/rfdetr-pose
Jun 17, 2026
Merged

feat: add RF-DETR keypoint preview model#1257
chmjkb merged 6 commits into
mainfrom
@chmjkb/rfdetr-pose

Conversation

@chmjkb

@chmjkb chmjkb commented Jun 17, 2026

Copy link
Copy Markdown
Collaborator

Register the RF-DETR keypoint preview pose model with xnnpack, coreml and mlx backends (all fp32). This is a beta preview export and may be re-exported under a different constant once a stable version ships.

  • modelUrls/modelRegistry: add the three backend URLs and variant map
  • PoseEstimationModule/types: register the model config (single-forward export, no inputSize axis) and extend PoseEstimationModelSources
  • demo: load it via usePoseEstimation in the pose estimation screen
  • docs: list it in the model registry and usePoseEstimation supported models

Description

Introduces a breaking change?

  • Yes
  • No

Type of change

  • Bug fix (change which fixes an issue)
  • New feature (change which adds functionality)
  • Documentation update (improves or adds clarity to existing documentation)
  • Other (chores, tests, code style improvements etc.)

Tested on

  • iOS
  • Android

Testing instructions

Screenshots

Related issues

Checklist

  • I have performed a self-review of my code
  • I have commented my code, particularly in hard-to-understand areas
  • I have updated the documentation accordingly
  • My changes generate no new warnings

Additional notes

@chmjkb chmjkb marked this pull request as ready for review June 17, 2026 12:28
@IgorSwat

Copy link
Copy Markdown
Contributor

One note: we should enable to pick the model in the demo app in the future, if we have 2 or more models for given type of task.

@chmjkb

chmjkb commented Jun 17, 2026

Copy link
Copy Markdown
Collaborator Author

One note: we should enable to pick the model in the demo app in the future, if we have 2 or more models for given type of task.

agreed, i'll open up another PR today :)

chmjkb and others added 3 commits June 17, 2026 15:02
Register the RF-DETR keypoint preview pose model with xnnpack, coreml and
mlx backends (all fp32). This is a beta preview export and may be
re-exported under a different constant once a stable version ships.

- modelUrls/modelRegistry: add the three backend URLs and variant map
- PoseEstimationModule/types: register the model config (single-`forward`
  export, no inputSize axis) and extend PoseEstimationModelSources
- demo: load it via usePoseEstimation in the pose estimation screen
- docs: list it in the model registry and usePoseEstimation supported models

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
@chmjkb chmjkb force-pushed the @chmjkb/rfdetr-pose branch from c4f013d to ae0c57d Compare June 17, 2026 13:03
@chmjkb chmjkb self-assigned this Jun 17, 2026
@chmjkb chmjkb added the model Issues related to exporting, improving, fixing ML models label Jun 17, 2026
@chmjkb chmjkb merged commit a53b536 into main Jun 17, 2026
5 checks passed
@chmjkb chmjkb deleted the @chmjkb/rfdetr-pose branch June 17, 2026 14:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

model Issues related to exporting, improving, fixing ML models

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants