作者实现的项目地址: https://github.com/IDEA-Research/DAB-DETR
项目中有两个模型
- DAB-DETR
- dab-deformable-detr (一个变体)
- matcher.py 文件,与Deformable DETR内容相同
- position_encoding.py 文件中多了一个PositionEmbeddingSineHW,对应于论文中的(Width & Height Modulated)
- attention.py 文件与Conditional DETR相同,移除了对QKV三者维度相同的检测,移除了内部的全连接,这些全连接放入了transformer网络中了
- transformer.py 主要不同的文件
- DABDETR.py 主要不同的文件