4 个月前

调和基础类和新类:一种用于广义少样本分割的类别对比方法

调和基础类和新类:一种用于广义少样本分割的类别对比方法

摘要

当前的少样本分割(FSSeg)方法主要集中在提高新类别的性能,而忽视了基础类别的性能。为了解决这一局限性,引入了广义少样本语义分割(GFSSeg)任务,旨在同时预测基础类别和新类别的分割掩模。然而,现有的基于原型的方法在更新原型时并未显式考虑基础类别与新类别之间的关系,导致在识别真实类别方面的性能有限。为了应对这一挑战,我们提出了一种类对比损失和一类关系损失来规范原型更新,并鼓励不同类别原型之间的较大距离,从而在区分各类别的同时保持基础类别的性能。我们的方法在PASCAL VOC和MS COCO数据集上实现了广义少样本分割任务的新最先进性能。

代码仓库

liuweide01/HBNC
官方
pytorch

基准测试

基准方法指标
generalized-few-shot-semantic-segmentation-on-2CCA (ResNet-50)
Mean Base and Novel: 27.86
Mean IoU: 37.48

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
调和基础类和新类:一种用于广义少样本分割的类别对比方法 | 论文 | HyperAI超神经