Segment Anything is A Good Pseudo-label Generator for Weakly Supervised Semantic Segmentation

Peng-Tao Jiang,Yuqi Yang
2023-05-02
Abstract:Weakly supervised semantic segmentation with weak labels is a long-lived ill-posed problem. Mainstream methods mainly focus on improving the quality of pseudo labels. In this report, we attempt to explore the potential of 'prompt to masks' from the powerful class-agnostic large segmentation model, segment-anything. Specifically, different weak labels are used as prompts to the segment-anything model, generating precise class masks. The class masks are utilized to generate pseudo labels to train the segmentation networks. We have conducted extensive experiments on PASCAL VOC 2012 dataset. Experiments demonstrate that segment-anything can serve as a good pseudo-label generator. The code will be made publicly available.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper attempts to address the problem of how to improve the quality of semantic segmentation with only weak labels (such as image-level labels, point labels, scribble labels, and bounding box labels). Traditional weakly supervised semantic segmentation methods mainly focus on improving the quality of pseudo-labels. In contrast, the authors of this paper explore the use of a powerful class-agnostic large-scale segmentation model—Segment Anything Model (SAM)—as a method for generating high-quality pseudo-labels. Specifically, different weak labels are used as prompts for SAM to generate precise class masks, which are further used to train the segmentation network. Experiments show that pseudo-labels generated using SAM can significantly improve the performance of semantic segmentation, achieving results on the PASCAL VOC 2012 dataset that are close to those obtained with finely annotated data. Additionally, the authors investigate the impact of different types of weak labels (such as image-level labels, point labels, scribble labels, and bounding box labels) on the quality of generated pseudo-labels and demonstrate the effectiveness of these different prompts in practical applications.