解决图像区域检测中的形状失真问题,提升开放词汇目标检测准确率
SIA-OVD: Shape-Invariant Adapter for Bridging the Image-Region Gap in Open-Vocabulary Detection
- 设计形状不变适配器SIA-OVD,动态选择不同形状区域的特征适配器
- 在COCO-OVD上显著提升区域分类准确率,优于现有主流方法
- 适合需要高精度开放词汇检测的视觉系统开发者
开放词汇目标检测(OVD)旨在无需实例级标注即可检测新物体,以降低开放世界目标检测的成本。现有方法主要依赖于如CLIP等视觉语言预训练模型的图文对齐能力,但这些模型训练于图像-文本对,缺乏对图像局部区域的感知能力,导致图像与区域表示之间存在差距。直接使用CLIP进行OVD会引发区域分类不准确。我们发现该差距主要源于感兴趣区域(RoI)提取过程中特征图的形变。为此,提出一种新的形状不变适配器SIA-OVD,学习不同形状区域的特征适配器,并设计新型适配器分配机制,为每个区域选择最优适配器。经适配后的区域表示能更贴近CLIP所学的文本表示。大量实验表明,SIA-OVD有效缓解了由形状变形引起的图像-区域间隙,显著提升区域分类精度,在COCO-OVD基准上实现显著超越现有方法的性能。代码已公开于https://github.com/PKU-ICST-MIPL/SIA-OVD_ACMMM2024。
原文摘要 · Abstract (English)
Open-vocabulary detection (OVD) aims to detect novel objects without instance-level annotations to achieve open-world object detection at a lower cost. Existing OVD methods mainly rely on the powerful open-vocabulary image-text alignment capability of Vision-Language Pretrained Models (VLM) such as CLIP. However, CLIP is trained on image-text pairs and lacks the perceptual ability for local regions within an image, resulting in the gap between image and region representations. Directly using CLIP for OVD causes inaccurate region classification. We find the image-region gap is primarily caused by the deformation of region feature maps during region of interest (RoI) extraction. To mitigate the inaccurate region classification in OVD, we propose a new Shape-Invariant Adapter named SIA-OVD to bridge the image-region gap in the OVD task. SIA-OVD learns a set of feature adapters for regions with different shapes and designs a new adapter allocation mechanism to select the optimal adapter for each region. The adapted region representations can align better with text representations learned by CLIP. Extensive experiments demonstrate that SIA-OVD effectively improves the classification accuracy for regions by addressing the gap between images and regions caused by shape deformation. SIA-OVD achieves substantial improvements over representative methods on the COCO-OVD benchmark. The code is available at https://github.com/PKU-ICST-MIPL/SIA-OVD_ACMMM2024.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。