一种基于轻量级脉冲SAM模型的图像分割方法

本发明公开了一种基于轻量级脉冲SAM模型的图像分割方法，包括以下步骤：获取待分割图像；构建学生模型，所述学生模型采用轻量级脉冲SAM模型；采用教师模型进行第一阶段的知识蒸馏训练，在该过程中对学生模型编码器输出的图像和教师模型输出的图像进行均方根误差计算；进行第二阶段的知识蒸馏训练，在该阶段使用二元交叉熵损失和DICE相似系数作为联合蒸馏损失，并对教师模型的软知识和学生模型的输出进行损失计算；以及表示硬知识的真值标签ytrue进行联合损失计算得到训练好的轻量级脉冲SAM模型；将待分割图像输入到训练好的轻量级脉冲SAM模型中，实现对图像的分割；本申请学生模型编码器具有参数量更少、体积更小、功耗更低的特点，因此更加适合于计算资源有限同时对能耗有更高要求的设备。