Anchor DETR: Query Design for Transformer-Based Detector
中文题目:
锚点DETR:基于transformer检测器的查询设计
论文链接:https://www.zhuanzhi.ai/paper/0700d0ebab68ec89aab99559175d9768
之前 transformer 检测算法的物体查询是一组可学向量,但这组向量没有显式的物理意义,查询对应的预测集也没有特定的模式因而难以优化。
本文提出基于锚点的查询设计,如此每个查询有显式的物理意义且仅关注锚点附近的物体。同时,本文设计的查询可以使一个锚点预测多个物体,解决一个区域可能有多个物体的难点。
本文还设计了一种行列特征解耦的 attention,减少显存的消耗且保持精度不降低。本方法实现简单,比 DETR 性能更高且优化更易、速度更快、显存更省。
专知便捷查看
便捷下载,请关注专知公众号(点击上方蓝色专知关注)
专知,专业可信的人工智能知识分发
,让认知协作更快更好!欢迎注册登录专知www.zhuanzhi.ai,获取5000+AI主题干货知识资料!
欢迎微信扫一扫加入专知人工智能知识星球群,获取最新AI专业干货知识教程资料和与专家交流咨询!
点击“
阅读原文
”,了解使用
专知
,查看获取5000+AI主题知识资源