Mar, 2022

利用潜在原型的对比增强学习进行少样本分割

TL;DR本文提出了一种利用潜在原型的对比增强方法来提高 few-shot 分割模型性能的方法,通过采样模块生成伪掩模和新的原型,从而提高原型和查询特征之间的相似性信息的利用率,并通过对比增强模块来提高预测性能。实验证明,该方法在 Pascal-5^i 和 COCO-20^i 的 1-shot 和 5-shot 分割任务中的性能明显优于现有方法,尤其在 5-shot 任务中,性能分别比基准提高了 5.9% 和 7.3%。