基于双重相似度孪生网络的小样本实例分割

作者：罗善威; 陈黎^*

来源：武汉科技大学学报, 2020, 43(01): 59-66.

DOI：10.3969/j.issn.1674-3644.2020.01.009

• 小样本学习
• 实例分割
• 孪生网络
• 残差网络
• 空域相似度
• 频域相似度

针对孪生网络在小样本数据集上的应用和优化问题,提出一种基于双重相似度计算和孪生网络的小样本实例分割模型。首先对传统孪生网络进行改进,将孪生网络与残差网络相结合,构建作为本模型骨干网络的孪生残差网络;然后在相似度计算阶段构建了具有两个子网络的双重相似度计算网络,分别用于计算场景图像与参考图像的空域相似度和频域相似度,并进行相似度特征融合;最后通过实例分割网络获得分割结果。此外,还引入Focal Loss损失函数来解决模型训练过程中正、负样本以及难、易样本的不均衡问题。在COCO数据集上的实验结果表明,本文方法的小样本实例分割性能要优于对比算法。

• 单位
  武汉科技大学; 智能信息处理与实时工业系统湖北省重点实验室