arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.11460cs.CV
收藏

RelTR:用于场景图生成的关系Transformer

RelTR: Relation Transformer for Scene Graph Generation

Leibniz University Hannover(汉诺威莱布尼茨大学);University of Twente(特文特大学)

Yuren Cong, Michael Ying Yang, Bodo Rosenhahn

AI总结:

本文提出端到端一阶段场景图生成模型RelTR,将关系预测建模为集合预测问题,通过编码器-解码器架构直接输出主语-谓词-宾语三元组,在Visual Genome和Open Images V6上取得优越性能与快速推理。

摘要:

同一场景中的不同对象彼此之间或多或少存在关联,但只有有限数量的这些关系值得关注。受在目标检测中表现优异的DETR启发,我们将场景图生成视为一个集合预测问题,并提出了一种端到端的场景图生成模型RelTR,该模型具有编码器-解码器架构。编码器对视觉特征上下文进行推理,而解码器利用不同类型的注意力机制,通过耦合的主语和宾语查询来推断固定大小的主语-谓词-宾语三元组集合。我们设计了一种集合预测损失,在端到端训练中执行真值三元组与预测三元组之间的匹配。与大多数现有场景图生成方法相比,RelTR是一种一阶段方法,仅使用视觉外观直接预测一组关系,而无需组合实体并标注所有可能的谓词。在Visual Genome和Open Images V6数据集上的大量实验证明了我们模型的优越性能和快速推理能力。

英文摘要:

Different objects in the same scene are more or less related to each other, but only a limited number of these relationships are noteworthy. Inspired by DETR, which excels in object detection, we view scene graph generation as a set prediction problem and propose an end-to-end scene graph generation model RelTR which has an encoder-decoder architecture. The encoder reasons about the visual feature context while the decoder infers a fixed-size set of triplets subject-predicate-object using different types of attention mechanisms with coupled subject and object queries. We design a set prediction loss performing the matching between the ground truth and predicted triplets for the end-to-end training. In contrast to most existing scene graph generation methods, RelTR is a one-stage method that predicts a set of relationships directly only using visual appearance without combining entities and labeling all possible predicates. Extensive experiments on the Visual Genome and Open Images V6 datasets demonstrate the superior performance and fast inference of our model.

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

↑