基于通用分割大模型的少样本大坝裂缝分割方法和系统

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
基于通用分割大模型的少样本大坝裂缝分割方法和系统
申请号:CN202410852379
申请日期:2024-06-28
公开号:CN118736226A
公开日期:2024-10-01
类型:发明专利
摘要
基于通用分割大模型的少样本大坝裂缝分割方法和系统,其方法包括:1)构建大坝裂缝分割数据集;2)训练裂缝图像分割器;3)编码有效提示信息;4)获取裂缝分割掩码。该方法结合传统的分割模型和通用分割大模型SAM的优点,在U‑Net架构的网络模型仅需要通过少量裂缝样本使其获得初步的裂缝识别能力,作为前置的通用裂缝分割器。随后将该训练完成的分割器获取的裂缝信息作为提示信息输入SAM的编码器中,通过编码器过滤无效信息整合有效信息获得提示信息的编码向量,和图像向量一起输入编码器中获取精确的裂缝掩码。本发明的方法克服了目前基于裂缝数据有监督训练的图像分割方法在样本不足的情况下往往难以达到最优效果且鲁棒性不高,进行迁移时效果不佳的问题,利用传统的分割模型和通用分割大模型SAM相结合,获取有效提示信息,提高了大坝裂缝分割方法的准确性、检出率和鲁棒性。
技术关键词
裂缝分割方法 图像分割器 大坝 图像块 裂缝特征 样本 前馈神经网络 注意力 图像编码器 分类方式 像素点 多尺度特征融合 联合损失函数 编码器特征 解码器