Can we visualize the intermediate result of Query Selection？

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"

Apache License 2.0

2.16k stars 234 forks source link

Can we visualize the intermediate result of Query Selection？ #83

Open TsingWei opened 2 years ago

TsingWei commented 2 years ago

I think it is fun to have the visualization on the top-k boxes selected from the Encoder's output.

FengLi-ust commented 2 years ago

Yes, you can just print the boxes of the selected boxes in query selection and visualize them. The quality of boxes after query selection is actually quite good.