DETR系列文章之–Efficient DETR
文章目录
前言
Anchor DETR发表于2022的AAAI,文章的主要思想是通过给每个可学习的Query Embedding赋予明确的物理意义,使其聚焦于特定的区域。
一、Anchor Points to Object Query
在原始DETR中,Attention公式如下:
本文作者是选用了两种类型的锚点:Grid Anchor Points和Learned Anchor Points,如下图所示:
Grid Anchor Points代码:
# 均匀网格锚点
nx=ny=round(math.sqrt(self.num_position))
self.num_position=nx*ny
x = (torch.arange(nx) + 0.5) / nx
y = (torch.arange(ny) + 0.5) / ny
xy=torch.meshgrid(x,y)