摘要
本发明公开了一种基于原型亲和力混合网络的小样本图像分割方法及系统,该方法包括以下步骤:对支持查询样本对进行特征提取,得到所述支持图像对应的支持特征和所述查询图像对应的查询特征;计算自适应查询原型和查询特征之间的相似度,得到保守查询预测掩码;利用原型引导特征增强,得到增强支持特征和增强查询特征;利用最后一个注意力块输出的基于注意力处理后的增强支持特征和注意力处理后的增强查询特征,生成最终查询预测掩码;利用损失函数对初始分割模型进行训练,得到目标分割模型;其能够实现全局语义引导与局部细节匹配的优势互补,提高模型在复杂场景下的分割精度与鲁棒性,满足对高精度像素级解析的需求。
技术关键词
查询特征
原型
图像分割方法
亲和力
混合网络
图像分割模型
注意力机制
样本
图像分割装置
校准
图像分割系统
模型训练模块
像素
图像获取模块
矩阵
特征提取模块
计算机程序产品