EGTR: Extracting Graph from Transformer for Scene Graph Generation

1 code implementation • 2 Apr 2024 • Jinbae Im, Jeongyeon Nam, Nokyung Park, HyungMin Lee, Seunghyun Park

We propose a lightweight one-stage SGG model that extracts the relation graph from the various relationships learned in the multi-head self-attention layers of the DETR decoder.

Graph Generation Multi-Task Learning +6