arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.10655cs.CV

OVLW-DETR:开放词汇轻量级检测Transformer

OVLW-DETR: Open-Vocabulary Light-Weighted Detection Transformer

Baidu Inc.(百度公司)

Yu Wang, Xiangbo Su, Qiang Chen, Xinyu Zhang, Teng Xi, Kun Yao, Errui Ding, Gang Zhang, Jingdong Wang

AI总结:

本研究提出部署友好的开放词汇轻量检测Transformer OVLW-DETR,通过替换检测器分类层权重为VLM文本编码器的类别名嵌入实现知识迁移,在Zero-Shot LVIS基准上性能优于现有实时开放词汇检测器。

摘要:

开放词汇目标检测聚焦于在自然语言引导下检测新类别。本报告提出开放词汇轻量级检测Transformer(OVLW-DETR),这是一款部署友好的开放词汇检测器,性能强劲且延迟低。基于OVLW-DETR,我们提供了一套端到端训练方案,通过简单对齐将视觉语言模型(VLM)的知识迁移到目标检测器。我们用从VLM的文本编码器中提取的类别名称嵌入,替换检测器中固定的分类层权重,以此实现检测器与文本编码器的对齐。OVLW-DETR无需额外融合模块,灵活性高且部署友好,更易于实现与调制,还提升了交错注意力计算的效率。实验结果表明,该方法在标准Zero-Shot LVIS基准上优于现有的实时开放词汇检测器。源代码与预训练模型可在https://github.com/Atten4Vis/LW-DETR 获取。

英文摘要:

Open-vocabulary object detection focusing on detecting novel categories guided by natural language. In this report, we propose Open-Vocabulary Light-Weighted Detection Transformer (OVLW-DETR), a deployment friendly open-vocabulary detector with strong performance and low latency. Building upon OVLW-DETR, we provide an end-to-end training recipe that transferring knowledge from vision-language model (VLM) to object detector with simple alignment. We align detector with the text encoder from VLM by replacing the fixed classification layer weights in detector with the class-name embeddings extracted from the text encoder. Without additional fusing module, OVLW-DETR is flexible and deployment friendly, making it easier to implement and modulate. improving the efficiency of interleaved attention computation. Experimental results demonstrate that the proposed approach is superior over existing real-time open-vocabulary detectors on standard Zero-Shot LVIS benchmark. Source code and pre-trained models are available at [https://github.com/Atten4Vis/LW-DETR].

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑