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
1.64k stars 178 forks source link

Why there are only 80 classes in `pred_logits`? #306

Closed pdt0609 closed 1 month ago

pdt0609 commented 1 month ago

Star RTDETR 请先在RTDETR主页点击star以支持本项目 Star RTDETR to help more people discover this project.

Dear author @lyuwenyu, I just have questions about the class classification of each queries: Why there are only 80 classes in pred_logits, while in DETR there remains 81 classes ( as background is counted as the 81st class). Why your model can still work well with COCO while not considering the "background" class? How RT-DETR can classify background class?

lyuwenyu commented 1 month ago

There is no background class. We use vfl as class loss.