跳到主要导航 跳到搜索 跳到主要内容

Few-shot object segmentation with a new feature aggregation module

  • Kaijun Liu
  • , Shujing Lyu*
  • , Palaiahnakote Shivakumara
  • , Yue Lu
  • *此作品的通讯作者
  • East China Normal University
  • University of Malaya

科研成果: 期刊稿件文章同行评审

摘要

The success of convolutional neural network for object segmentation depends on a large amount of training data and high-quality samples. But annotating such high-quality training data for pixel-wise segmentation is labor-intensive. To reduce the massive labor work, few-shot learning has been introduced to segment objects, which uses a few samples for training without compromising the performance. However, the current few-shot models are biased towards the seen classes rather than being class-irrelevant due to lack of global context prior attention. Therefore, this study aims at proposing a few-shot object segmentation model with a new feature aggregation module. Specifically, the proposed work develops a detail-aware module to enhance the discrimination of details with diversified attributes. To enhance the semantics of each pixel, we propose a global attention module to aggregate detailed features containing semantic information. Furthermore, to improve the performance of the proposed model, the model uses support samples that represents class-specific prototype obtained by respective category prototype block. Next, the proposed model predicts label of each pixel of query sample by estimating the distance between the pixel and prototypes. Experiments on standard datasets demonstrate significance of the proposed model over SOTA in terms of segmentation with a few training samples.

源语言英语
文章编号102459
期刊Displays
78
DOI
出版状态已出版 - 7月 2023

学术指纹

探究 'Few-shot object segmentation with a new feature aggregation module' 的科研主题。它们共同构成独一无二的学术指纹。

引用此