基于自动提示学习的目标实例分割模型建立方法及其应用

Abstract

本发明公开了基于自动提示学习的目标实例分割模型建立方法及其应用，属于图像分割领域，包括：建立目标分割网络，并利用已标注地面真值标签的图像数据集对其进行训练；目标分割网络包括：特征嵌入网络，其包括SAM中的图像编码器，用于获取输入图像的图像特征嵌入；轮廓映射识别网络，根据图像特征嵌入预测各目标的类别、中心点位置以及中心点到n个轮廓点的距离；自动提示学习网络，用于筛选出最重要的部分轮廓点与中心点一起映射到语义空间，得到点提示的语义嵌入；以及掩码解码器，其为SAM中的掩码解码器，用于根据图像特征嵌入和点提示的语义嵌入生成分割掩码。本发明能够生成高质量的、可靠的点提示嵌入，提高图像分割的实用性和泛化性。

Images