Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Self-Support Few-Shot Semantic Segmentation

About

Existing few-shot segmentation methods have achieved great progress based on the support-query matching framework. But they still heavily suffer from the limited coverage of intra-class variations from the few-shot supports provided. Motivated by the simple Gestalt principle that pixels belonging to the same object are more similar than those to different objects of same class, we propose a novel self-support matching strategy to alleviate this problem, which uses query prototypes to match query features, where the query prototypes are collected from high-confidence query predictions. This strategy can effectively capture the consistent underlying characteristics of the query objects, and thus fittingly match query features. We also propose an adaptive self-support background prototype generation module and self-support loss to further facilitate the self-support matching procedure. Our self-support network substantially improves the prototype quality, benefits more improvement from stronger backbones and more supports, and achieves SOTA on multiple datasets. Codes are at \url{https://github.com/fanq15/SSP}.

Qi Fan, Wenjie Pei, Yu-Wing Tai, Chi-Keung Tang• 2022

Related benchmarks

TaskDatasetResultRank
Few-shot Semantic SegmentationPASCAL-5^i (test)--
177
Few-shot SegmentationCOCO 20^i (test)
mIoU44.1
174
Semantic segmentationCOCO-20i
mIoU (Mean)44.1
132
Few-shot Semantic SegmentationCOCO 5-shot 20i
mIoU50.2
85
Few-shot SegmentationPASCAL 5i (val)
mIoU (Mean)69.3
83
Few-shot Semantic SegmentationCOCO-20i (test)
mIoU (mean)44.1
79
Few-shot SegmentationCOCO-20^i--
78
Few-shot Semantic SegmentationCOCO 20i 1-shot
mIoU (Overall)42
77
Few-shot Semantic SegmentationFSS-1000
mIoU79.4
64
Few-shot SegmentationDeepGlobe
mIoU54.2
61
Showing 10 of 26 rows

Other info

Code

Follow for update