Skip to content

Add support for RF-DETR base model from HuggingFace (Roboflow) #4

@headertag

Description

@headertag

Is your feature request related to a problem? Please describe.
I want to use the RF-DETR base model from HuggingFace (Roboflow) in this repository to leverage modern transformer-based object detection capabilities.

Describe the solution you'd like
Integrate the RF-DETR base model available at https://huggingface.co/Roboflow/rf-detr-base using the HuggingFace Transformers API. Provide code/documentation for:

  • Model loading from HuggingFace
  • Inference pipeline setup
  • Integration examples within the repo's workflows

Describe alternatives you've considered

  • Integrating other object detection models instead of RF-DETR
  • Manual/standalone implementation without HuggingFace Hub

Additional context
Official model card: https://huggingface.co/Roboflow/rf-detr-base
The model supports both PyTorch and ONNX formats.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions