← 返回列表
一种图像分割方法、系统、存储介质和装置
摘要文本
本发明涉及一种图像分割方法、系统、存储介质和装置。主要包括:获取待分类的图像数据;采用分割模型对图像数据进行处理,以获得图像数据的分类结果,其中,分割模型包括一个教师模型和一个学生模型,所述分割模型是通过以下方法进行训练:对训练过程中的学生模型参数取指数移动平均,以获得更加鲁棒的教师模型,并用教师模型的预测结果作为伪标签来监督学生模型的学习。该图像分割方法中,训练过程中的学生模型参数取指数移动平均,从而获得更加鲁棒的教师模型,并用教师模型的预测结果作为伪标签来监督学生模型的学习。有利于提高图像分割方法的准确度。
申请人信息
- 申请人:江苏新希望科技有限公司
- 申请人地址:215000 江苏省苏州市姑苏区长洲路吏舍弄10号6号楼
- 发明人: 江苏新希望科技有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 一种图像分割方法、系统、存储介质和装置 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311405974.3 |
| 申请日 | 2023/10/27 |
| 公告号 | CN117333874A |
| 公开日 | 2024/1/2 |
| IPC主分类号 | G06V20/70 |
| 权利人 | 江苏新希望科技有限公司 |
| 发明人 | 左严; 杨萍萍; 汤斌; 王正荣; 王祥伟; 包寅杰; 张世兰; 刘振威 |
| 地址 | 江苏省苏州市姑苏区长洲路吏舍弄10号6号楼 |
专利主权项内容
1.一种图像分割方法,其特征在于,包括:获取待分类的图像数据;采用分割模型对图像数据进行处理,以获得图像数据的分类结果,其中,分割模型包括一个教师模型和一个学生模型,所述分割模型是通过以下方法进行训练:所述学生模型的参数以表示,所述教师模型的参数以/>表示,学生模型使用指数移动平均算法来更新教师模型的参数, 具体应用以下公式:
其中t表示相应的训练步骤,/>,/>为相应的平滑系数,/>用来调整/>和/>的权重,对于样本对于样本x,学生模型对其预测结果为,教师模型对其预测结果为/>,/>和/>可以通过以下公式获得:
,
,其中,学生模型和教师模型采用相同模型结构,使用表示,其中/>和/>为不同的高斯噪声,将教师模型的输出作为伪标签,以此监督学生模型学习,最终损失函数为:,其中y表示样本x所对应的真实标签,/>为分割损失函数,/>为一致性损失函数,/>用以控制和/>的比重。通过最终损失函数对所述分割模型进行训练。