首页 | 本学科首页   官方微博 | 高级检索  
     


Few-shot object segmentation with a new feature aggregation module
Affiliation:1. State Key Laboratory of Virtual Reality Technology and Systems, Beihang University, Beijing 100191, China;2. School of Computer Science and Engineering, Beihang University, Beijing 100191, China;3. Schoolof Computer Scienceand Technology,Civil Aviation University of China, Tianjin 300300, China
Abstract: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.
Keywords:Few-shot learning  Object segmentation  Feature aggregation  Attention mechanism
本文献已被 ScienceDirect 等数据库收录!
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号