一种基于局部注意力网络的街景图像语义分割方法

摘要

本发明公开的一种基于局部注意力网络的街景图像语义分割方法，具体实施步骤是：步骤1，首先从公开数据集Cityscapes中随机选取部分图像数据，将选取的部分图像数据划分为训练集、验证集及测试集；步骤2，使用倒置残差模块和空洞卷积构造MobileNetV2网络模型；步骤3，设计局部注意力模块和残差块，并构造编码网络；步骤4，构造解码网络，逐渐恢复图像分辨率，最终输出语义分割的结果；步骤5，使用训练集和验证集对模型进行训练，并在测试集上验证模型的分割效果。该方法解决了现有技术中存在的普遍存在的局部信息在特征提取过程中无法完整保留，从而解决了类别内部分割结果不一致的问题。