基于注意力机制的图像语义分割特征融合方法

摘要

本发明公开了一种基于注意力机制的图像语义分割特征融合方法，步骤如下：采集样本图像，利用卷积神经网络提取特征，获取样本图像的深层特征图和浅层特征图，调整特征图，将调整后的深层特征图和浅层特征图输入到注意力机制模块；在注意力模块中，将深层特征图进行降维，再将深层特征图聚类得到的语义特征分配到浅层特征图中，将特征图通道数恢复后输出融合特征图；将步骤2输出的融合特征图和步骤1调整之后的浅层特征图进行融合，之后再和步骤1调整后的深层特征图进行融合，得到最后融合特征图。本发明采用注意力机制辅助特征融合，将深层网络提取的语义特征与浅层网络保留的细节特征进行高效融合，提升了语义分割网络的分割精度。