4 个月前

CoDet:共现引导的区域-词汇对齐用于开放词汇对象检测

CoDet:共现引导的区域-词汇对齐用于开放词汇对象检测

摘要

从图像-文本对中提取可靠的区域-词汇对齐对于学习面向开放词汇的对象检测的物体级视觉-语言表示至关重要。现有方法通常依赖于预训练或自训练的视觉-语言模型进行对齐,这些方法容易在定位精度或泛化能力上存在局限性。本文提出了一种新颖的方法——CoDet,该方法通过将区域-词汇对齐重新表述为共现对象发现问题,克服了对预对齐视觉-语言空间的依赖。直观上,通过将带有包含相同概念的字幕的图像分组,对应于该共享概念的对象在组内应表现出较高的共现率。CoDet随后利用视觉相似性来发现这些共现对象,并将其与共享概念对齐。大量实验表明,CoDet在开放词汇检测方面具有优越的性能和令人信服的可扩展性,例如,在扩大视觉主干网络规模后,CoDet在OV-LVIS数据集上分别达到了37.0 $\text{AP}^m_{\text{novel}}$和44.7 $\text{AP}^m_{\text{all}}$,超过了之前的最先进水平(SoTA)4.2 $\text{AP}^m_{\text{novel}}$和9.8 $\text{AP}^m_{\text{all}}$。代码可在https://github.com/CVMI-Lab/CoDet获取。

代码仓库

cvmi-lab/codet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
open-vocabulary-object-detection-on-lvis-v1-0CoDet (EVA02-L)
AP novel-LVIS base training: 37.0

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CoDet:共现引导的区域-词汇对齐用于开放词汇对象检测 | 论文 | HyperAI超神经