← 返回列表

一种图像分割方法、系统、存储介质和装置

申请号: CN202311405974.3
申请人: 江苏新希望科技有限公司
申请日期: 2023/10/27

摘要文本

本发明涉及一种图像分割方法、系统、存储介质和装置。主要包括:获取待分类的图像数据;采用分割模型对图像数据进行处理,以获得图像数据的分类结果,其中,分割模型包括一个教师模型和一个学生模型,所述分割模型是通过以下方法进行训练:对训练过程中的学生模型参数取指数移动平均,以获得更加鲁棒的教师模型,并用教师模型的预测结果作为伪标签来监督学生模型的学习。该图像分割方法中,训练过程中的学生模型参数取指数移动平均,从而获得更加鲁棒的教师模型,并用教师模型的预测结果作为伪标签来监督学生模型的学习。有利于提高图像分割方法的准确度。

专利详细信息

项目 内容
专利名称 一种图像分割方法、系统、存储介质和装置
专利类型 发明申请
申请号 CN202311405974.3
申请日 2023/10/27
公告号 CN117333874A
公开日 2024/1/2
IPC主分类号 G06V20/70
权利人 江苏新希望科技有限公司
发明人 左严; 杨萍萍; 汤斌; 王正荣; 王祥伟; 包寅杰; 张世兰; 刘振威
地址 江苏省苏州市姑苏区长洲路吏舍弄10号6号楼

专利主权项内容

1.一种图像分割方法,其特征在于,包括:获取待分类的图像数据;采用分割模型对图像数据进行处理,以获得图像数据的分类结果,其中,分割模型包括一个教师模型和一个学生模型,所述分割模型是通过以下方法进行训练:所述学生模型的参数以表示,所述教师模型的参数以/>表示,学生模型使用指数移动平均算法来更新教师模型的参数, 具体应用以下公式:
其中t表示相应的训练步骤,/>,/>为相应的平滑系数,/>用来调整/>和/>的权重,对于样本对于样本x,学生模型对其预测结果为,教师模型对其预测结果为/>,/>和/>可以通过以下公式获得:

,其中,学生模型和教师模型采用相同模型结构,使用表示,其中/>和/>为不同的高斯噪声,将教师模型的输出作为伪标签,以此监督学生模型学习,最终损失函数为:,其中y表示样本x所对应的真实标签,/>为分割损失函数,/>为一致性损失函数,/>用以控制和/>的比重。通过最终损失函数对所述分割模型进行训练。