search

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.1k stars 230 forks source link

Visualize bbox for all decoder layers #182

Closed akankshasingh25 closed 1 year ago

akankshasingh25 commented 1 year ago

Hey,

I am trying to evaluate DINO on my custom dataset; how can we visualize the bounding boxes predicted after every decoder of DINO?

Thanks

FengLi-ust commented 1 year ago

Hi, the output contains predictions from all decoder layers. You can visualize the one you want by specify the output prediction layer.