-
公开(公告)号:CN119205801A
公开(公告)日:2024-12-27
申请号:CN202411248458.9
申请日:2024-09-06
Applicant: 厦门大学
IPC: G06T7/10
Abstract: 一种基于SAM模型的快速且显存友好的图像分割方法,所述图像分割方法包括如下步骤:第一步,基于SAM模型训练建立U‑SAM模型;该U‑SAM模型采用如下步骤进行训练:步骤S1:建立SAM模型和U‑SAM模型;通过知识蒸馏压缩SAM模型的图像编码器,得到U‑SAM模型的语义&尺度感知图像编码器,以提升SAM的语义感知能力;步骤S2:冻结SAM模型相关参数,保留可提示分割能力,训练U‑SAM模型的双子令牌掩码解码器,解决难以控制提示嵌入存在的问题。第二步,利用所述U‑SAM模型实现可提示分割和通用图像分割;由此,可通过一个快速且显存友好的框架实现提示和通用的分割任务。