-
Notifications
You must be signed in to change notification settings - Fork 462
Add RFDetr model for Object Detection and Instance Segmentation #5081
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: develop
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull request overview
This PR adds support for the RF-DETR (Real-time Fast DETR) object detection model from Roboflow, integrating it into the OTX framework. RF-DETR combines a DINOv2 backbone with a lightweight DETR decoder for real-time object detection.
Key changes:
- Integration of RF-DETR model with five variants (nano, small, base, medium, large)
- Unit tests for RF-DETR model and detector wrapper
- Training recipes for all model variants
Reviewed changes
Copilot reviewed 11 out of 11 changed files in this pull request and generated 2 comments.
Show a summary per file
| File | Description |
|---|---|
| library/src/otx/backend/native/models/detection/rfdetr.py | Core RF-DETR model implementation with training, inference, and export functionality |
| library/src/otx/backend/native/models/detection/detectors/rfdetr.py | Detector wrapper that interfaces between OTX and the rfdetr package |
| library/src/otx/backend/native/models/detection/detectors/init.py | Added RFDETRDetector export |
| library/src/otx/backend/native/models/detection/init.py | Added RFDETR model export |
| library/src/otx/recipe/detection/rfdetr_*.yaml | Training configuration files for all five RF-DETR variants |
| library/tests/unit/backend/native/models/detection/test_rfdetr.py | Unit tests for RFDETR model class |
| library/tests/unit/backend/native/models/detection/detectors/test_rfdetr_detector.py | Unit tests for RFDETRDetector wrapper |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
Summary
How to test
Checklist