Abstract:Weakly supervised semantic segmentation is receiving great attention due to its low human annotation cost. In this paper, we aim to tackle bounding box supervised semantic segmentation, i.e., training accurate semantic segmentation models using bounding box annotations as supervision. To this end, we propose Affinity Attention Graph Neural Network ($A^2$GNN). Following previous practices, we first generate pseudo semantic-aware seeds, which are then formed into semantic graphs based on our newly proposed affinity Convolutional Neural Network (CNN). Then the built graphs are input to our $A^2$GNN, in which an affinity attention layer is designed to acquire the short- and long- distance information from soft graph edges to accurately propagate semantic labels from the confident seeds to the unlabeled pixels. However, to guarantee the precision of the seeds, we only adopt a limited number of confident pixel seed labels for $A^2$GNN, which may lead to insufficient supervision for training. To alleviate this issue, we further introduce a new loss function and a consistency-checking mechanism to leverage the bounding box constraint, so that more reliable guidance can be included for the model optimization. Experiments show that our approach achieves new state-of-the-art performances on Pascal VOC 2012 datasets (val: 76.5\%, test: 75.2\%). More importantly, our approach can be readily applied to bounding box supervised instance segmentation task or other weakly supervised semantic segmentation tasks, with state-of-the-art or comparable performance among almot all weakly supervised tasks on PASCAL VOC or COCO dataset. Our source code will be available at <a class="link-external link-https" href="https://github.com/zbf1991/A2GNN" rel="external noopener nofollow">this https URL</a>.

Graph-Boosted Convolutional Neural Networks For Semantic Segmentation

Class-related Graph Convolution for Weakly Supervised Semantic Segmentation

CNN-G: Convolutional Neural Network Combined With Graph for Image Segmentation With Theoretical Analysis

Weakly-Supervised Semantic Segmentation Network with Iterative Dcrf

Weakly Supervised Semantic Segmentation with Boundary Exploration

Learning to Exploit the Prior Network Knowledge for Weakly-Supervised Semantic Segmentation

Affinity Attention Graph Neural Network for Weakly Supervised Semantic Segmentation

A survey of semi- and weakly supervised semantic segmentation of images

Group-Wise Semantic Mining for Weakly Supervised Semantic Segmentation

Weakly Supervised Semantic Segmentation Based on Co-segmentation.

Semantic Segmentation by Improved Generative Adversarial Networks

Semi-supervised Semantic Segmentation via Strong-Weak Dual-Branch Network

On Boosting Semantic Street Scene Segmentation with Weak Supervision

Weakly Supervised Semantic Segmentation by Multiple Group Cosegmentation

Enhanced Pseudo-Label Generation with Self-supervised Training for Weakly-supervised Semantic Segmentation

Weakly Supervised Semantic Segmentation Based on Web Image Co-segmentation

An Attention Enhanced Graph Convolutional Network for Semantic Segmentation

Self-constructing graph neural networks to model long-range pixel dependencies for semantic segmentation of remote sensing images

A Regularized Convolutional Neural Network for Semantic Image Segmentation

Superpixel Guided Network for Weakly Supervised Semantic Segmentation

Bottom-Up Top-Down Cues for Weakly-Supervised Semantic Segmentation