Twostage deformable detr and dab

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

Twostage deformable detr and dab #5

Closed Ixiaohuihuihui closed 2 years ago

Ixiaohuihuihui commented 2 years ago

Can't two stage and use DAB be set to true at the same time? if self.two_stage: ...

elif self.use_dab: ...

SlongLiu commented 2 years ago

As shown in our paper, we did not test the two-stage variant of our DAB-Deformable-DETR.

SlongLiu commented 2 years ago

Even so, our DAB-Deformable-DETR has outperformed the two-stage Deformable DETR with a large margin. A new PR to support two-stage DAB-Deformable-DETR is welcome.