IDEA-Research / DN-DETR

[CVPR 2022 Oral] Official implementation of DN-DETR
Apache License 2.0
530 stars 57 forks source link

How Known Labels Detection implemented? #5

Closed fblogy closed 2 years ago

fblogy commented 2 years ago

Thanks for your excellent work. Could you give more details how Known Labels Detection implemented?How do you let the decoder output all boxes of specific class c only using the label embedding of class c?

HaoZhang534 commented 2 years ago

You can output boxes from all classes and only select those from the known class.

fblogy commented 2 years ago

Are you replace the all unknown class embedding by label embedding of class c in the matching part ,and the number of times you need to inference is the number of classes in the image?

HaoZhang534 commented 2 years ago

Yes, your understanding is right.

tzhhhh123 commented 1 year ago

Thanks for your excellent work. May I ask about the specific implementation of the fine-tuning phase?