xiuqhou / Relation-DETR

[ECCV2024 Oral] Official implementation of the paper "Relation DETR: Exploring Explicit Position Relation Prior for Object Detection"
Apache License 2.0
81 stars 6 forks source link

与Co-DETR的对比与结合以及大规模预训练 #1

Open Artificial-Inability opened 1 month ago

Artificial-Inability commented 1 month ago

Question

您好,感谢您出色的工作!Relation DETR和Co DETR都是可以在DINO的基础上通过加入即插即用的模块提升较多性能,COCO 1x性能分别能达到51.7和52.1,请问你们有尝试过在Co DETR的框架基础上加入你们提出的模块吗,性能是否有进一步提升呢? 另一个问题是论文中仅提供了r50和swinL直接在COCO上训练的性能,Co-DETR使用VitL主干经过大规模预训练后在COCO上能达到66.0的性能,你们是否做了大规模预训练(如Obj365)后再在COCO上训练的实验或者用更大的主干训练的实验呢,性能怎么样?

补充信息

No response

xiuqhou commented 1 month ago

Hi,感谢你的提问,这两个问题真是问到关键点上了!

第一个问题

  1. 我们自己并没有试过Co-DETR和relation的结合,主要是Co--ETR本身结合FastrRCNN和ATSS就已经比较复杂了。我更倾向于将Co-DETR作为一个成熟的sota方法,而不是一个更好的baseline。
  2. 另外RelationDETR是在我自己构建的一套框架中实现的,为了将relation与其他方法结合,需要重新整理实现其他方法的代码。本身我还有科研任务,可能没时间去一一尝试将relation与各种方法结合,尤其是像Co-DETR这种特别复杂的方法。
  3. 后续我出一个文档,说明如何将我们的方法与现有模型结合,就像DN-DETR提出的denoising机制那样,欢迎有兴趣的研究者自己探索我们的方法是否对Co-DETR有效!

第二个问题,其实我有想过冲一冲COCO-leaderboard,看看RelationDETR到底性能咋样,奈何组里只有8张3090+2张A800,实在没法跑O365预训练🤣我手头有一个结果可以供你参考:

显然这只是个次优结果,如果能够和Co DETR保持相同的实验条件:

我觉得是有机会达到和Co DETR接近的AP的。

除了性能,我们的优势在于其简单通用,没有在DINO上引入太多代码处理,很适合作为一个更好的baseline去进行更多扩充,例如引入rank-detr中的排序机制、ms-detr中基于阈值的一对多匹配,甚至也可以仿照CoDETR将FasterRCNN和ATSS模块引入进来。换言之他应该是有进一步的提升潜力。