Skip to content
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

Request to add DINO object detector #36205

Open
2 tasks done
tcourat opened this issue Feb 14, 2025 · 4 comments
Open
2 tasks done

Request to add DINO object detector #36205

tcourat opened this issue Feb 14, 2025 · 4 comments

Comments

@tcourat
Copy link

tcourat commented Feb 14, 2025

Model description

DINO (do not confuse it with the DINO image encoder from META) is a SOTA DETR-like object detector, improving the denoising training, query initialization, and box prediction. It is based on a combination of the enhancement brought by DN-DETR , DAB-DETR , and Deformable DETR.
As it is used as backbone for many other DETR architecture (e.g Co-DETR which is SOTA on COCO test-dev : https://paperswithcode.com/sota/object-detection-on-coco), it would be nice to have it in transformers.

Additionnaly, a slighly improved version of DINO, called Stable-DINO, also exist, and should be easily added on top of DINO (only a few lines of code).

Open source status

  • The model implementation is available
  • The model weights are available

Provide useful links for the implementation

Paper : https://arxiv.org/abs/2203.03605
Code : https://github.com/IDEA-Research/DINO

Code for Stable-DINO : https://github.com/IDEA-Research/Stable-DINO

@qubvel
Copy link
Member

qubvel commented Feb 15, 2025

Hi @court thanks for opening the issue! That would be a great addition to our object-detection series, would you like to work on it, or should we seek anyone who will take this up?

@tcourat
Copy link
Author

tcourat commented Feb 15, 2025

Hi @tcourat thanks for opening the issue! That would be a great addition to our object-detection series, would you like to work on it, or should we seek anyone who will take this up?

I do not plan to work on it at the moment as I am not familiar with HF integration practices, but since I believe this would be a great addition to HF I created this request.

@Ruhaan838
Copy link

Hi, @tcourat @qubvel. If you don't mind, I need to contribute to transformers. Can I do that?

@qubvel
Copy link
Member

qubvel commented Feb 16, 2025

Hi @Ruhaan838, sure, we would appreciate your contribution 🤗 Feel free to ping me when PR is open

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants