Pixel-wise Reclassification with Prototypes for Enhancing Weakly Supervised Semantic Segmentation

Yujie Diao,Ruiguo Yu,Xuewei Li,Yilong Fan,Zhiqiang Liu,Mei Yu,Chenhan Wang,Jie Gao
DOI: https://doi.org/10.3233/faia240491
2024-01-01
Abstract:Refining the seed region to obtain finely annotated pseudo masks for training a segmentation model is a crucial step in the multi-stage weakly supervised semantic segmentation (WSSS) framework. One of the most popular refinement methods, IRN, extends seed regions towards the edges in the image. However, we observed that, due to the lack of guidance from semantic information, IRN’s refinement may lead the generation of partially erroneous refinement directions. To address this issue, we leverage prototypes to recover the overlooked category semantic information in the refinement stage. We propose a prototype-based pseudo mask reclassification post-processing (PtReCl) to correct misclassified pixels in the pseudo masks, generating refined pseudo masks with more accurate coverage. Experimental evaluations demonstrate that our post-processing approach brings improvements in both pseudo mask quality and segmentation results on PASCAL VOC and MS COCO datasets, achieving state-of-the-art performance on VOC.
What problem does this paper attempt to address?