Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Towards Flexible Visual Relationship Segmentation

About

Visual relationship understanding has been studied separately in human-object interaction(HOI) detection, scene graph generation(SGG), and referring relationships(RR) tasks. Given the complexity and interconnectedness of these tasks, it is crucial to have a flexible framework that can effectively address these tasks in a cohesive manner. In this work, we propose FleVRS, a single model that seamlessly integrates the above three aspects in standard and promptable visual relationship segmentation, and further possesses the capability for open-vocabulary segmentation to adapt to novel scenarios. FleVRS leverages the synergy between text and image modalities, to ground various types of relationships from images and use textual features from vision-language models to visual conceptual understanding. Empirical validation across various datasets demonstrates that our framework outperforms existing models in standard, promptable, and open-vocabulary tasks, e.g., +1.9 $mAP$ on HICO-DET, +11.4 $Acc$ on VRD, +4.7 $mAP$ on unseen HICO-DET. Our FleVRS represents a significant step towards a more intuitive, comprehensive, and scalable understanding of visual relationships.

Fangrui Zhu, Jianwei Yang, Huaizu Jiang• 2024

Related benchmarks

TaskDatasetResultRank
Human-Object Interaction DetectionHICO-DET (test)
mAP (full)38.1
493
Human-Object Interaction DetectionV-COCO (test)
AP (Role, Scenario 1)65.2
270
Human-Object Interaction DetectionHICO-DET (Rare First Unseen Combination (RF-UC))
mAP (Full)36.6
77
Human-Object Interaction DetectionHICO-DET Non-rare First Unseen Composition (NF-UC)
AP (Unseen)26.62
49
HOI DetectionHICO-DET (test)
Box mAP (Full)38.1
32
Human-Object Interaction DetectionV-COCO
Box mAP (Scenario 1)65.2
32
Human-Object Interaction DetectionHICO-DET Unseen Composition (UC)
Unseen AP14.48
17
HOI SegmentationHICO-DET (test)
mask mAP (Full)40.5
12
Human-Object Interaction SegmentationHICO-DET (test)
mask mAP (Full)40.5
12
Human-Object Interaction DetectionHICO-DET Unseen Verb (UV)
Unseen Score21.5
11
Showing 10 of 13 rows

Other info

Code

Follow for update