Visual Relationship Detection with Relative Location Mining
This work addresses the challenge of detecting interactions between objects in images, which is important for scene understanding in computer vision, but it appears incremental as it builds on existing methods by incorporating location-based features.
The paper tackles visual relationship detection by mining relative location information to improve object-pair proposing and predicate recognition, achieving state-of-the-art results on VRD and VG datasets.
Visual relationship detection, as a challenging task used to find and distinguish the interactions between object pairs in one image, has received much attention recently. In this work, we propose a novel visual relationship detection framework by deeply mining and utilizing relative location of object-pair in every stage of the procedure. In both the stages, relative location information of each object-pair is abstracted and encoded as auxiliary feature to improve the distinguishing capability of object-pairs proposing and predicate recognition, respectively; Moreover, one Gated Graph Neural Network(GGNN) is introduced to mine and measure the relevance of predicates using relative location. With the location-based GGNN, those non-exclusive predicates with similar spatial position can be clustered firstly and then be smoothed with close classification scores, thus the accuracy of top $n$ recall can be increased further. Experiments on two widely used datasets VRD and VG show that, with the deeply mining and exploiting of relative location information, our proposed model significantly outperforms the current state-of-the-art.