论文标题

类别感知的变压器网络,用于更好的人类对象互动检测

Category-Aware Transformer Network for Better Human-Object Interaction Detection

论文作者

Dong, Leizhen, Li, Zhimin, Xu, Kunlun, Zhang, Zhijun, Yan, Luxin, Zhong, Sheng, Zou, Xu

论文摘要

人类对象相互作用(HOI)检测旨在在识别其相互作用的同时定位人类和相关对象,对于理解静止图像至关重要。最近,基于变压器的模型显着提高了HOI检测的进展。但是,由于模型的对象查询总是简单地将其简单地作为零来初始化,因此这些模型的功能尚未得到充分探索,这将影响性能。在本文中,我们试图通过使用类别感知的语义信息初始化对象查询来研究促进基于变压器的HOI检测器的问题。为此,我们创新提出了类别感知的变压器网络(CATN)。具体而言,将通过以外部对象检测模型为代表的类别先验初始化对象查询,以产生更好的性能。此外,此类先验可以进一步用于通过注意机制增强特征的表示能力。我们首先通过Oracle实验验证了我们的想法,通过使用地面图类别信息初始化对象查询。然后进行了广泛的实验,以表明配备了我们想法的HOI检测模型超过基线,以实现新的最新结果。

Human-Object Interactions (HOI) detection, which aims to localize a human and a relevant object while recognizing their interaction, is crucial for understanding a still image. Recently, transformer-based models have significantly advanced the progress of HOI detection. However, the capability of these models has not been fully explored since the Object Query of the model is always simply initialized as just zeros, which would affect the performance. In this paper, we try to study the issue of promoting transformer-based HOI detectors by initializing the Object Query with category-aware semantic information. To this end, we innovatively propose the Category-Aware Transformer Network (CATN). Specifically, the Object Query would be initialized via category priors represented by an external object detection model to yield better performance. Moreover, such category priors can be further used for enhancing the representation ability of features via the attention mechanism. We have firstly verified our idea via the Oracle experiment by initializing the Object Query with the groundtruth category information. And then extensive experiments have been conducted to show that a HOI detection model equipped with our idea outperforms the baseline by a large margin to achieve a new state-of-the-art result.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源