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.

Sign up for GitHub

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

Jump to bottom

[Object Detection] Add DETR Model #2043

Draft

DavidLandup0 wants to merge 5 commits into keras-team:master from DavidLandup0:detr

+628 −0

Collaborator

DavidLandup0 commented Jan 14, 2025

Draft PR for observability. Body will be updated as progress is made.

DETR Elements

Will include:

DETR layers
- Positional embeddings
- Frozen BatchNormalization
- DETRTransformerEncoder
- DETRTransformerDecoder
DetrBackbone (Modified ResNet)
Object detection preprocessor
DETRObjectDetector (Tying the above into a class)
Weight conversion script from https://huggingface.co/facebook/detr-resnet-50

... and relevant workflows, collab notebooks, etc. for sanity checking.

DavidLandup0 added 5 commits

January 14, 2025 23:22


          Initial commit - first couple of layers

3ed6014


          Add transformer layers - trimmed down

b103664


          remove cache/decoding steps

cef86cb


          Convert einsumdense into dense layers, remove output range

8f33663


          Slight cleanup

1b0b97d

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

Labels

None yet