IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
https://detrex.readthedocs.io/en/latest/
Apache License 2.0
1.99k stars 206 forks source link

关于mixed_selection的问题 #331

Closed nhw649 closed 9 months ago

nhw649 commented 9 months ago

作者你好,h_deformable_detr中使用的mixed_selection是哪篇文章提出的呢?是Conditional DETR for Fast Training Convergence这篇的idea还是一个提点的trick呢?

nhw649 commented 9 months ago
if not self.mixed_selection:
    query_pos, query = torch.split(pos_trans_out, c, dim=2)
else:
    # query_pos here is the content embed for deformable DETR
    query = query_embed.unsqueeze(0).expand(bs, -1, -1)
    query_pos, _ = torch.split(pos_trans_out, c, dim=2)
rentainhe commented 9 months ago

是DINO这篇工作里提到的一个涨点技巧,可以参考一下:

nhw649 commented 9 months ago

是DINO这篇工作里提到的一个涨点技巧,可以参考一下:

好的,谢谢。