search

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Apache License 2.0
2.15k stars 232 forks source link

Mixed Query Selection #153

Open liangjxiong opened 1 year ago

liangjxiong commented 1 year ago

您好！我刚入门深度学习，我想知道在您的代码中，有关仅仅实现Mixed Query Selection的代码在哪部分呢？可以指出来吗？我很着急的想知道。非常感谢您杰出的工作！

HaoZhang534 commented 1 year ago

https://github.com/IDEA-Research/DINO/blob/main/models/dino/transformer_deformable.py#L178-L204

liangjxiong commented 1 year ago

在topk_proposals = torch.topk(enc_outputs_class[..., 0], topk, dim=1)[1]中，为什么是选择第0类enc_outputs_class[..., 0]前topk个？期待您的回复！