search

IDEA-Research / DN-DETR

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

How to calculate flops #44

Closed stnjumu closed 1 year ago

stnjumu commented 1 year ago

Hi! thanks for your excellent work, I'm wondering how to evaluate the flops of DN-DETR model? I can't easily use the DETR script below because of the dn_components. https://github.com/facebookresearch/detr/issues/110 Could you pls share your python script?

stnjumu commented 1 year ago

and I notice that the GFLOPs of DAB-DETR and DN-DETR are the same, and the eval mode doesn't need dn_components. Maybe I can calculate the Flops of DAB-DETR instead?

FengLi-ust commented 1 year ago

Yes, our gflops is the same as DAB-DETR and you can calculate the gflops with DAB-DETR script.

stnjumu commented 1 year ago

ok, thanks! I'll ask them.