Open liangjxiong opened 1 year ago
您好!我刚入门深度学习,我想知道在您的代码中,有关仅仅实现Mixed Query Selection的代码在哪部分呢?可以指出来吗?我很着急的想知道。非常感谢您杰出的工作!
https://github.com/IDEA-Research/DINO/blob/main/models/dino/transformer_deformable.py#L178-L204
在topk_proposals = torch.topk(enc_outputs_class[..., 0], topk, dim=1)[1]中,为什么是选择第0类enc_outputs_class[..., 0]前topk个?期待您的回复!
您好!我刚入门深度学习,我想知道在您的代码中,有关仅仅实现Mixed Query Selection的代码在哪部分呢?可以指出来吗?我很着急的想知道。非常感谢您杰出的工作!