[发明专利]语义分割方法、装置和系统及存储介质在审
| 申请号: | 201810333109.5 | 申请日: | 2018-04-13 |
| 公开(公告)号: | CN108876793A | 公开(公告)日: | 2018-11-23 |
| 发明(设计)人: | 章圳黎;张祥雨;彭超 | 申请(专利权)人: | 北京迈格威科技有限公司 |
| 主分类号: | G06T7/11 | 分类号: | G06T7/11;G06K9/62 |
| 代理公司: | 北京睿邦知识产权代理事务所(普通合伙) 11481 | 代理人: | 徐丁峰;戴亚南 |
| 地址: | 100190 北京市海*** | 国省代码: | 北京;11 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 语义分割 待处理图像 上采样模块 存储介质 输入特征 特征输入 同一通道 同一位置 网络结构 网络输出 元素变换 网络 | ||
1.一种语义分割方法,包括:
获取待处理图像;
将所述待处理图像输入U型网络,以获得所述U型网络输出的所述待处理图像的语义分割结果,其中,所述U型网络的扩张路径包括至少一个上采样模块,在所述至少一个上采样模块的每一个中,将输入特征中分别位于p个通道上的同一位置处的p个元素变换为同一通道上的相邻的p个元素,以获得变换后的特征,并将所述变换后的特征输入后续网络结构,其中,n是大于1的整数,1≤i<n。
2.如权利要求1所述的方法,其中,所述U型网络的收缩路径包括依次连接的n个卷积模块,所述将所述待处理图像输入U型网络,以获得所述U型网络输出的所述待处理图像的语义分割结果包括:
将所述待处理图像输入所述n个卷积模块中的第一个卷积模块;
对于所述n个卷积模块中的第m个至第n个卷积模块中的至少一个卷积模块中的每一个,
将该卷积模块的输出特征输入至该卷积模块对应的跳跃连接模块;
将所述跳跃连接模块的输出特征输入至该卷积模块对应的上采样模块以获得所述变换后的特征,或将所述跳跃连接模块的输出特征输入至该卷积模块对应的合并模块以获得合并后的特征,并将所述合并后的特征输入至该卷积模块对应的上采样模块以获得所述变换后的特征,其中,该卷积模块对应的上采样模块为所述至少一个上采样模块之一;
将所述变换后的特征输入在该卷积模块之前的上一卷积模块所对应的合并模块,在所述合并模块中,将所述变换后的特征与在该卷积模块之前的上一卷积模块所对应的跳跃连接模块的输出特征进行合并,并将合并后的特征输入后续网络结构,所述后续网络结构用于输出所述待处理图像的语义分割结果,
其中,n是大于1的整数,1<m<n,p是大于1的整数。
3.如权利要求2所述的方法,其中,所述方法还包括:
获取训练图像以及对应的分割标注数据,所述分割标注数据用于指示所述训练图像中的每个像素属于至少一个预定类别的概率;
将所述训练图像输入所述U型网络;
对于所述至少一个卷积模块中的每一个,
将该卷积模块对应的上采样模块的输出特征输入到该卷积模块对应的辅助监督模块,以获得所述辅助监督模块输出的所述训练图像的语义分割结果;
基于所述训练图像的语义分割结果与所述分割标注数据计算该卷积模块对应的分类损失;
基于所述至少一个卷积模块对应的分类损失计算总的损失;
基于所述总的损失优化所述U型网络,以获得经训练的所述U型网络。
4.如权利要求3所述的方法,其中,所述辅助监督模块包括依次连接的大小调整模块、通道调整模块和第一分类函数层,其中,
所述大小调整模块用于将所述上采样模块的输出特征的大小调整到与所述训练图像的大小一致,所述通道调整模块用于将所述大小调整模块的输出特征的通道数调整到预定数目,所述第一分类函数层用于基于所述通道调整模块的输出特征预测所述训练图像中的每个像素属于所述至少一个预定类别的概率。
5.如权利要求4所述的方法,其中,所述大小调整模块包括至少一个卷积层和/或至少一个池化层,和/或所述通道调整模块包括至少一个卷积层。
6.如权利要求2所述的方法,其中,对于所述至少一个卷积模块中的每一个,该卷积模块对应的后续网络结构包括至少一个反卷积模块、在最后一个反卷积模块之后依次连接的密集邻近预测模块和第二分类函数层,其中,
所述最后一个反卷积模块的输出特征的通道划分为k*k组,每个组中的第q个通道用于预测同一类别,其中,k为大于1的整数,q为大于0的整数;
所述密集邻近预测模块用于对所述最后一个反卷积模块的输出特征求平均,获得平均特征;
所述第二分类函数用于基于所述平均特征计算所述待处理图像的语义分割结果。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京迈格威科技有限公司,未经北京迈格威科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201810333109.5/1.html,转载请声明来源钻瓜专利网。





