[发明专利]语义分割模型的半监督学习方法、相关装置及产品在审
申请号: | 202110691567.8 | 申请日: | 2021-06-22 |
公开(公告)号: | CN113379750A | 公开(公告)日: | 2021-09-10 |
发明(设计)人: | 李莹莹;蒋旻悦;戴欣怡;谭啸;孙昊 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
主分类号: | G06T7/10 | 分类号: | G06T7/10;G06K9/38;G06K9/46;G06K9/62;G06N3/04;G06N3/08 |
代理公司: | 北京英赛嘉华知识产权代理有限责任公司 11204 | 代理人: | 王达佐;马晓亚 |
地址: | 100085 北京市*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 语义 分割 模型 监督 学习方法 相关 装置 产品 | ||
1.一种语义分割模型的半监督学习方法,执行如下半监督学习操作,直至得到所述语义分割模型:
从训练样本集中选取训练样本,其中,所述训练样本集中包括带标签的训练样本和不带标签的训练样本;
基于所述训练样本,得到第一图像和第二图像;
通过初始语义分割模型,得到所述第一图像的第一分割结果和所述第二图像的第二分割结果;
确定所述第一分割结果和所述第二分割结果之间的第一一致性损失,以及语义分割过程中,相对应的所述第一图像的第一目标特征和所述第二图像的第二目标特征之间的第二一致性损失;
根据所述第一一致性损失和所述第二一致性损失,调整所述初始语义分割模型。
2.根据权利要求1所述的方法,其中,所述第一目标特征包括所述第一分割结果中表征目标对象的边缘的关键点特征,所述第二目标特征包括所述第二分割结果中表征目标对象的边缘的关键点特征;以及
确定语义分割过程中,相对应的所述第一图像的第一目标特征和所述第二图像的第二目标特征之间的第二一致性损失,包括:
提取所述第一分割结果中的目标对象的关键点特征和所述第二分割结果中的目标对象的关键点特征;
根据所述第一分割结果和所述第二分割结果中的、相对应的关键点特征,确定所述第二一致性损失。
3.根据权利要求1-2中任一项所述的方法,其中,所述第一目标特征包括所述初始语义分割模型对于所述第一图像的语义分割过程所得到的第一中间特征,所述第二目标特征包括所述初始语义分割模型对于所述第二图像的语义分割过程所得到的、对应于所述第一中间特征的第二中间特征;以及
确定语义分割过程中,相对应的所述第一图像的第一目标特征和所述第二图像的第二目标特征之间的第二一致性损失,包括:
确定所述第一中间特征和所述第二中间特征之间的第二一致性损失。
4.根据权利要求1所述的方法,其中,所述基于所述训练样本,得到第一图像和第二图像,包括:
以所述训练样本作为所述第一图像,在所述训练样本中加入噪声,得到所述第二图像。
5.根据权利要求4所述的方法,其中,还包括:
响应于确定所述训练样本为带标签的训练样本,确定所述训练样本的标签与所述第一分割结果之间的第三损失;以及
所述根据所述第一一致性损失和所述第二一致性损失,调整所述初始语义分割模型,包括:
根据所述第一一致性损失、所述第二一致性损失和所述第三损失,调整所述初始语义分割模型。
6.一种语义分割方法,包括:
获取待分割图像;
通过语义分割模型处理所述待分割图像,得到分割结果,其中,所述语义分割模型根据权利要求1-5的方法训练得到。
7.一种语义分割模型的半监督学习装置,通过如下单元执行半监督学习操作,直至得到所述语义分割模型:
选取单元,被配置成从训练样本集中选取训练样本,其中,所述训练样本集中包括带标签的训练样本和不带标签的训练样本;
得到单元,被配置成基于所述训练样本,得到第一图像和第二图像;
分割单元,被配置成通过初始语义分割模型,得到所述第一图像的第一分割结果和所述第二图像的第二分割结果;
第一确定单元,被配置成确定所述第一分割结果和所述第二分割结果之间的第一一致性损失,以及语义分割过程中,相对应的所述第一图像的第一目标特征和所述第二图像的第二目标特征之间的第二一致性损失;
调整单元,被配置成根据所述第一一致性损失和所述第二一致性损失,调整所述初始语义分割模型。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202110691567.8/1.html,转载请声明来源钻瓜专利网。