Anchor DETR: Query Design for Transformer-Based Detector
中文题目:锚点DETR:基于transformer检测器的查询设计 论文链接:https://www.zhuanzhi.ai/paper/0700d0ebab68ec89aab99559175d9768
之前 transformer 检测算法的物体查询是一组可学向量,但这组向量没有显式的物理意义,查询对应的预测集也没有特定的模式因而难以优化。
本文提出基于锚点的查询设计,如此每个查询有显式的物理意义且仅关注锚点附近的物体。同时,本文设计的查询可以使一个锚点预测多个物体,解决一个区域可能有多个物体的难点。
本文还设计了一种行列特征解耦的 attention,减少显存的消耗且保持精度不降低。本方法实现简单,比 DETR 性能更高且优化更易、速度更快、显存更省。