【CVPR2020-港科大-腾讯】基于注意力机制及多关系检测器的小样本物体检测

本文由香港科技大学和腾讯优图实验室联合提出。目标检测网络已经被广泛应用到安保，自动驾驶，医学图像等各个领域。然而传统的目标检测网络需要使用大量高质量的训练样本对模型进行训练。这些训练样本需要大量的人力物力进行标注，往往无法快速获得，所以无法将目标检测模型快速部署到新样本的检测中，而小样本目标检测方法可以很好地解决这一问题。我们提出了一种基于深度孪生网络的小样本目标检测模型，通过基于注意力机制的候选框网络，多关系检测器以及三元组对比训练方法对网络进行改进，使得网络能够不对新物体重新训练即可应用于新类别检测。此外，我们提供了一个1000类的小样本物体检测数据集，希望可以方便该领域的研究。

我们的工作主要有以下贡献：

首先，我们使用注意力机制对物体检测候选框进行筛选。我们将待检测新物体的特征作为滤波器在输入图片上进行卷积，以此找出潜在的候选框区域。

然后，我们使用多关系检测器对这些候选框进行分类以及位置调整。多关系检测器对候选框和新物体进行像素级、区域级和全图级的多级关系匹配，以此找出匹配程度最高的区域作为检测输出。

最后，我们构建（目标样本，正样本，负样本）训练样本三元组对模型进行训练，使得网络能够同时学习到相同物体间的相似性和不同物体间的差异性，从而大大提升网络在新样本上的检测性能。我们的方法在多个数据集上均取得了最好的结果，且无需在新物体上进行任何训练。其基本框架图如下：

https://www.zhuanzhi.ai/paper/8574d53bcbd28edd4c889bf3d4c1fd16

专知便捷查看

便捷下载，请关注专知公众号（点击上方蓝色专知关注）

• 后台回复“ARPN” 就可以获取《【CVPR2020-港科大-腾讯】基于注意力机制及多关系检测器的小样本物体检测》专知下载链接

专知，专业可信的人工智能知识分发，让认知协作更快更好！欢迎注册登录专知www.zhuanzhi.ai，获取5000+AI主题干货知识资料！

欢迎微信扫一扫加入专知人工智能知识星球群，获取最新AI专业干货知识教程资料和与专家交流咨询！

点击“ 阅读原文 ”，了解使用 专知 ，查看获取5000+AI主题知识资源