search

IDEA-Research / DAB-DETR

[ICLR 2022] Official implementation of the paper "DAB-DETR: Dynamic Anchor Boxes are Better Queries for DETR"
Apache License 2.0
501 stars 86 forks source link

How to use the swin-transformer as the backbone for DAB-DETR #23

Open yuan738 opened 2 years ago

yuan738 commented 2 years ago

Thank you for help!

yuan738 commented 2 years ago

When I use swin-transformer as the backbone of the model, the error is ValueError: return_layers are not present in model.

So how can I modify the code of return_layers in backbone.py?

Thank you.

SaifeiYan commented 2 years ago

When I use swin-transformer as the backbone of the model, the error is ValueError: return_layers are not present in model.

So how can I modify the code of return_layers in backbone.py?

Thank you. do you solve this problem,can you help me

yuan738 commented 2 years ago

sry, I didn't solve this problem.

fernandorovai commented 2 years ago

I'm trying to fix this one as well

rentainhe commented 1 year ago

Thank you for help!

Hi~ We've released our new codebase for DETR-like models called detrex, we've already supported Swin-Backbone for DAB-DETR in detrex now. And we've released new baselines with better mAP for DAB-DETR in detrex.

By using detrex, it's easier for you to use other backbones instead of Swin, ResNet, etc. @SaifeiYan @yuan738 @fernandorovai