PRNet: prototype reorganization few-shot semantic segmentation network
摘要
Few-shot semantic segmentation aims to address the dependence of traditional models on large-scale datasets, hoping to learn a model that can accurately segment unseen classes in a query image using only a few annotated samples (support images). Existing methods still face challenges in handling intra-class variations and in utilizing information. To alleviate these issues, we propose the Prototype Reorganization Network (PRNet). It includes innovative designs in three aspects: (1) A simple and efficient network framework that not only retains generalization ability but also improves the utilization of high-level information through a novel prior generation method. (2) A novel Prototype Reconstruction Transformer that overcomes the information loss issue inherent in traditional prototype generation methods. (3) A Learnable Prototype Fusion (LPF) strategy that can fully utilize category information and effectively mitigate the interference of intra-class variations and noise. Extensive experiments demonstrate PRNet’s superior performance on benchmark datasets such as PASCAL-