首页/ 人民网股份有限公司/ 专利详情

基于通道注意力的图像语义分割方法及装置

基本信息

申请号 CN202110837049.2 申请日 -
公开（公告）号 CN113361537A 公开（公告）日 2021-09-07
申请公布号 CN113361537A 申请公布日 2021-09-07
分类号 G06K9/34;G06K9/46;G06N3/04;G06N3/08 分类 计算；推算；计数;
发明人 郭俊波;郭筱凤;靳国庆;马凌峰;谢洪涛;张勇东 申请（专利权）人 人民网股份有限公司
代理机构 北京市浩天知识产权代理事务所（普通合伙） 代理人 宋菲
地址 100026 北京市西城区新街口外大街28号B座234号
法律状态 -

摘要

摘要 本发明公开了一种基于通道注意力的图像语义分割方法及装置，该方法包括：对原始图像进行预处理，得到待分割的输入图；将输入图输入至图像语义分割模型的特征提取网络中，利用特征提取网络对输入图进行特征提取，得到特征图；其中，特征提取网络的各个阶段的各个瓶颈模块中插入有通道注意力模块，用于将输入特征经过水平池化和竖直池化，得到两个矩阵；从通道角度使用1维卷积并行处理两个矩阵，并进行归一化处理后取平均值，得到通道权重；将特征图输入至图像语义分割模型的解码器中，利用解码器对特征图进行处理，得到预测分割图。本发明是通过通道注意力机制改进编码器的性能，能够很方便地移植到各种分割网络中，能够有明显的性能提升。