megvii-research / AnchorDETR

An official implementation of the Anchor DETR.
Other
338 stars 36 forks source link

How to get the 151 GFLOPs? #22

Closed charlesCXK closed 2 years ago

charlesCXK commented 2 years ago

Hi, thank you for doing such nice work! May I ask how to get the number of GFLOPs in Table 1? I got 172G using the same strategy of DETR (average on the first 100 images of COCO val set). Thanks!

charlesCXK commented 2 years ago

I use FlopCountAnalysis in detectron2, which has different rules with the tool in this repo.

tangjiuqi097 commented 2 years ago

Thanks for your reminder. We follow the script in the #110 of DETR with a few modifications. We will update the repo and the paper at the next version by using the FlopCountAnalysis in detectron2, as we think it is more widely used.