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

    公开(公告)号:CN118279320A

    公开(公告)日:2024-07-02

    申请号:CN202410365151.0

    申请日:2024-03-28

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

    基于边缘信息增强的图像实例检测分割模型构建方法

    公开(公告)号:CN118762042A

    公开(公告)日:2024-10-11

    申请号:CN202410736505.8

    申请日:2024-06-07

    Abstract: 本发明属于图像实例检测分割技术领域,公开了一种基于边缘信息增强的图像实例检测分割模型构建方法,包括:采用数据集训练实例检测分割网络,得到实例检测分割模型;实例检测分割网络包括:特征提取模块,用于提取每个输入样本图像不同尺度的高、低分辨率特征图,构成特征金字塔;边缘信息增模块,对输入样本图像进行边缘信息增强,得到边缘信息增强的特征图;多尺度特征融合模块,用于将边缘信息增强的特征图以及特征金字塔中的其他特征图进行多尺度自适应融合,得到融合后的特征图;预测模块,用于基于融合后的特征图进行目标的实例类别预测、实例的边界框位置预测,以及二值掩模预测。本发明能够提升图像实例检测分割的准确度。

Patent Agency Ranking