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.31k stars 258 forks source link

关于论文the uncertainty minimal query selection使用编码器的结果没有体现在vfl loss里面 #357

Open r4hul77 opened 2 months ago

r4hul77 commented 2 months ago

Hello Firstly Thanks for rt-detr, I was really curious about how the query selection and where query selection is done and I believe it is done in vfl loss. As per my intuition the query selection has to be done using the encoder outputs $\hat{X}$, but the vfl loss just uses decoder outputs aka "pred_boxes" and "pred_logits". Can you please tell me what am I missing from the code ?