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.16k stars 234 forks source link

save_results #71

Closed abbasmammadov closed 2 years ago

abbasmammadov commented 2 years ago

I have tried --save_results feature with several datasets, but it triggers an issue that outputs['pred_boxes'][0] has a shape of (900, 4) but results[0]['scores'] and results[0]['labels'] have shapes of (300). Since save_results concatenates those by unsqueezeing latter 2 on dim=-1, the error occurs here. I tried with several models and custom datasets but the error looks inevitable. Can you please consider this issue? Thanks in advance!

Screen Shot 2022-08-11 at 6 55 51 PM
HaoZhang534 commented 2 years ago

Hi @abbasmammadov , I'm sorry for the bug. Actually, although there is a "save_results" option, we never used it. We may support "save_results" in the next version of our code.

abbasmammadov commented 2 years ago

Ohh, I see. Thank you very much for your clear and prompt reply :)