search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.16k stars 233 forks source link

The questions of IoU-aware Query Selection #347

Open Yyc1999super opened 2 months ago

Yyc1999super commented 2 months ago

前辈，请问论文中提到的IoU-aware Query Selection的模块实现代码是在哪里呢？论文中描述的是在Encoder和Decoder之间，但是在代码中没有体现出来（也可能是我太菜了，看不明白），我看到了_get_decoder_input这个函数，但是觉得没有相关的操作。可否指点一下，麻烦您了。

lyuwenyu commented 2 months ago

在loss里 具体实现采用的vfl

Yyc1999super commented 2 months ago

好的，感谢

Jenny199099 commented 1 week ago

楼主你好，请问你找到IoU-aware Query Selection的模块实现代码了吗？能否分享一下，谢谢。

Yyc1999super commented 1 week ago

楼主你好，请问你找到IoU-aware Query Selection的模块实现代码了吗？能否分享一下，谢谢。

在rtdetr_criterion.py里面的loss_labels_vfl函数。

1999zsh commented 14 hours ago

楼主你好，我看论文里面没有说IoU-aware Query Selection，而是uncertainty minimal query selection方法。您指的是后者吗？还有我发现这个方法的代码与论文里面不是很一致呀

img