基于提示学习的小样本缺陷分割方法及系统

本发明提供了基于提示学习的小样本缺陷分割方法及系统，涉及图像识别技术领域，包括，获取待测缺陷图像，已知缺陷图像以及已知缺陷图像的缺陷掩码图；分别提取待测缺陷图像和已知缺陷图像的多维度特征；根据待测缺陷图像和已知缺陷图像的多维度特征，获取待测缺陷图像和已知缺陷图像的全局相似特征和局部相似特征，合并全局相似特征和局部相似特征，得到聚合相似度特征；基于聚合相似度特征，分别提取稠密提示向量和稀疏提示向量；将稠密提示向量和稀疏提示向量以及待测缺陷图像的多维度特征输入掩膜解码器，得到待测缺陷图像的缺陷分割结果。本发明能够提高小样本缺陷分割的精度。