Pyramid Graph Networks with Connection Attentions for Region-BasedOne-Shot Semantic Segmentation

Page view(s)
26
Checked on May 20, 2024
Pyramid Graph Networks with Connection Attentions for Region-BasedOne-Shot Semantic Segmentation
Title:
Pyramid Graph Networks with Connection Attentions for Region-BasedOne-Shot Semantic Segmentation
Journal Title:
The IEEE International Conference on Computer Vision (ICCV), 2019
DOI:
Publication Date:
03 November 2019
Citation:
Abstract:
One-shot image segmentation aims to undertake the segmentation task of a novel class with only one training image available. The difficulty lies in that image segmentation has structured data representations, which yields a many-to-many message passing problem. Previous meth-ods often simplify it to a one-to-many problem by squeezing support data to a global descriptor. However, a mixed global representation drops the data structure and information of individual elements. In this paper, We propose to model structured segmentation data with graphs and apply attentive graph reasoning to propagate label information from support data to query data. The graph attention mechanism could establish the element-to-element correspondence across structured data by learning attention weights between connected graph nodes. To capture correspondence at different semantic levels, we further propose a pyramid-like structure that models different sizes of image regions as graph nodes and undertakes graph reasoning at different levels. Experiments on PASCAL VOC 2012 dataset demonstrate that our proposed network significantly outperforms the baseline method and leads to new state-of-the-art performance on 1-shot and 5-shot segmentation benchmarks.
License type:
PublisherCopyrights
Funding Info:
Description:
ISBN:

Files uploaded:
File Size Format Action
There are no attached files.