Skip to content

Add LWDetr model#40991

Merged
Cyrilvallez merged 85 commits intohuggingface:mainfrom
sbucaille:add_lw_detr
Jan 12, 2026
Merged

Add LWDetr model#40991
Cyrilvallez merged 85 commits intohuggingface:mainfrom
sbucaille:add_lw_detr

Conversation

@sbucaille
Copy link
Contributor

What does this PR do?

Adds LWDetr model.
In #36895 I started working on adding RFDetr, but after putting some work I realized that it relies a LOT on LWDetr.
Adding RFDetr will essentially replace the ViT encoder by Dino, so the biggest part of the work is on the implementation of LWDetr, which could also be a good alternative for people to use for their use cases.

Who can review?

Still work in progress but since @yonigozlan asked for an update here it is.
All the inference code is implemented. A lot of refactoring/renaming is still needed and I'm writing the tests to be able to do that safely. In the meantime you can check the code and let me know if you have comments.

@qubvel

Loading
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants