Few-shot segmentation network based on class-aware prototype fusion
摘要
Most existing few-shot segmentation methods based on the support-query matching framework suffer from insufficient support information, where the limited number and coverage of annotated samples often produce prototypes that are incomplete or contaminated by background features, leading to incomplete activation of target regions in the query image and false activation of irrelevant areas. To address this issue, we propose a class-aware prototype fusion network (CAPFN) for few-shot segmentation, comprising a class-aware module (CAM) and a prototype fusion module (PFM). The CAM extracts class-specific information by jointly utilizing support features, masks, and preliminary query predictions, thereby guiding the model to attend precisely to target regions. To further mitigate semantic gap between the support and query domains, the PFM constructs a hybrid prototype by fusing support-derived and query-derived prototypes based on initial predictions. This fusion enhances prototype quality, reduces information loss, and improves the discriminative capacity of query activation. Extensive experiments on benchmark datasets (PASCAL-5