arXiv:2411.10261cs.CV2024-11TPAMI被引 3

解决图像中局部文字片段的精准检索难题。

Partial Scene Text Retrieval

  • 通过共享特征空间实现查询文本与场景文字的跨模态匹配。
  • 在无额外标注情况下,准确检索出文字行内的局部片段。
  • 提出动态匹配算法,提升局部检索效率和精度。

部分场景文字检索任务旨在从图像库中定位并搜索与给定查询文本相同或相似的文字实例。现有方法仅能处理文字行级实例,无法检索文字行内部的局部片段,因训练数据缺乏片段级标注。为此,本文提出一种可同时检索文字行及其局部片段的网络。该方法将查询文本与场景文字实例嵌入共享特征空间,并计算跨模态相似度。为处理局部片段,采用多实例学习(MIL)机制,无需额外标注即可学习其与查询文本的相似性。但传统MIL构建样本袋会引入大量噪声样本,降低训练效率与推理速度。为此,本文提出排序型多实例学习(RankMIL),自适应过滤噪声样本。此外,设计动态局部匹配算法(DPMA),可在推理阶段直接从文字行中搜索目标局部片段,无需构建样本袋,显著提升检索效率与性能。代码与数据集见:https://github.com/lanfeng4659/PSTR。

原文摘要 · Abstract (English)

The task of partial scene text retrieval involves localizing and searching for text instances that are the same or similar to a given query text from an image gallery. However, existing methods can only handle text-line instances, leaving the problem of searching for partial patches within these text-line instances unsolved due to a lack of patch annotations in the training data. To address this issue, we propose a network that can simultaneously retrieve both text-line instances and their partial patches. Our method embeds the two types of data (query text and scene text instances) into a shared feature space and measures their cross-modal similarities. To handle partial patches, our proposed approach adopts a Multiple Instance Learning (MIL) approach to learn their similarities with query text, without requiring extra annotations. However, constructing bags, which is a standard step of conventional MIL approaches, can introduce numerous noisy samples for training, and lower inference speed. To address this issue, we propose a Ranking MIL (RankMIL) approach to adaptively filter those noisy samples. Additionally, we present a Dynamic Partial Match Algorithm (DPMA) that can directly search for the target partial patch from a text-line instance during the inference stage, without requiring bags. This greatly improves the search efficiency and the performance of retrieving partial patches. The source code and dataset are available at https://github.com/lanfeng4659/PSTR.

文字检索局部匹配多实例学习跨模态

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。