(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211159243.0 (22)申请日 2022.09.22 (71)申请人 合肥工业大 学 地址 230000 安徽省合肥市屯溪路193号 (72)发明人 林勇　钟乐天　林学威　苏羿安　 方贤宝　金钊　 (74)专利代理 机构 合肥铭辉知识产权代理事务 所(普通合伙) 34212 专利代理师 张名列 (51)Int.Cl. G06V 20/10(2022.01) G06V 10/26(2022.01) G06V 10/80(2022.01) G06V 10/82(2022.01) (54)发明名称 一种基于改进Unet网络模型的遥感图像道 路分割方法 (57)摘要 本发明涉及遥感图像识别技术领域， 公开了 一种基于改进Unet网络模型的遥感图像道路分 割方法， 所述方法包括： 将Unet训练模型中的卷 积层修改为ResNet残差模 块， 添加了改进后的空 洞空间金字塔池化模块融合不同尺度的信息， 并 对编码部分添加一部分的跳跃连接， 从而缓解部 分上采样丢失信息； 并对训练样本进行数据增 强， 归一化处理； 将处理后的图像数据引入训练 好的改进Unet模型， 得到分割后的结果， 得到遥 感中的道路信息。 本发明在U net原始网络模型的 基础上增加改进， 添加了改进的ASPP模块， 使得 道路分割精度得到较大提高， 可实现遥感图像道 路信息的实时精准分割。 权利要求书1页 说明书3页 附图2页 CN 115512220 A 2022.12.23 CN 115512220 A 1.一种基于改进 Unet网络模型的遥感图像道路分割方法， 其特 征在于， 所述方法包括： 步骤一、 将Unet训练模型中的卷积层修改为ResNet残差模块， 添加改进后的空洞 空间 金字塔池化模块融合不同尺度的信息， 并对编码部分添加跳跃连接， 缓解上采样的信息丢 失； 步骤二、 对训练样本进行 数据增强， 归一 化处理； 步骤三、 将处理后的图像数据引入训练好的改进Unet模型， 得到分割后的结果， 最终获 取遥感中的道路信息 。 2.根据权利要求1所述的一种基于改进Unet网络模型的遥感图像道路分割方法， 其特 征在于， 步骤一中， 所述改进后的空洞空间金字塔池化模块介于解码网络与编码网络之间 添加， 所述跳跃 连接在解码网络中添加。 3.根据权利要求1所述的一种基于改进Unet网络模型的遥感图像道路分割方法， 其特 征在于， 步骤二和步骤三中， 具体步骤 包括： S1、 采集遥感图像， 并对遥感图像进行图像预处 理； S2、 对预处 理后的图像进行 标注， 将标注后的图像调整为统一尺寸， 制作训练数据集； S3、 将得到的训练数据集作为改进 Unet模型的输入图像数据， 训练改进 Unet分割模型； S4、 将采集的图像数据输入训练好的改进Unet模型之中， 通过训练好的改进Unet模型 获得分割后的结果， 最终 获取遥感中的道路信息 。 4.根据权利要求3所述的一种基于改进Unet网络模型的遥感图像道路分割方法， 其特 征在于， 所述S2中， 对预处理后的图像进行标注， 将标注后的图像调整为统一尺寸， 其具体 过程为： 对增强预处理后的图像进行标注， 获得label形式的遥感图像， 并将标注后获得的 图像调整为512 ×512的PNG图片。 5.根据权利要求1所述的一种基于改进Unet网络模型的遥感图像道路分割方法， 其特 征在于， 所述改进 Unet网络模型的损失函数L 为： 其公式中， GT代表标签图Ground  Truth， PR代表预测图 Predicti on， GT2和PR2分别表示两样本各自的元 素总数。权　利　要　求　书 1/1 页 2 CN 115512220 A 2一种基于改进U net网络模型的遥感图像道路分割方 法 技术领域 [0001]本发明涉及遥感图像识别技术领域， 尤其涉及一种基于改进Unet网络模型的遥感 图像道路分割方法。 背景技术 [0002]随着科技的发展， 获取遥感图像的成本越来越低， 导致各个遥感平台可以获得比 之前更多的遥感数据。 常见的遥感数据包括： 高中低分辨率数据， 高光谱 数据， 雷达数据， 激 光数据等。 而在遥感数据中， 通常遥感图像中道路和其他背 景特征元素复杂多样， 所以对其 进行语义分割对道路信息进行特征提取仍然存在精度等局限问题。 而 大多数仍然停留在人 为的目视识别层面上， 通常只是把遥感图像当做基础信息， 并没有对其进行信息的提取与 处理。 但是随着深度学习技术的发展与加深， 利用深度学习来对遥感 图像中目标信息进行 提取分类意 义变得更加重要。 [0003]传统的图像分割方法有阈值分割法， 边缘分割法， 区域分割法等。 但是遥感图像含 有丰富信息， 目标尺度复杂等特点， 并不能像简单图像一样 被传统方法所分割。 并且传统图 像分割方法所得出的最 终结果没有语义标注。 而近年来， 随着深度学习的不断发展， 神经网 络强大的特征提取能力， 基于深度学习的语义分割模型逐渐成为了主流遥感图像分割方 法。 不仅在精度还是速度上都有了明显提升 。 发明内容 [0004]本发明的目的是为了解决现有技术中存在的缺点， 而提出的一种基于改进Unet网 络模型的遥感图像道路分割方法。 [0005]一种基于改进 Unet网络模型的遥感图像道路分割方法， 所述方法包括： [0006]步骤一、 将Unet训练模型中的卷积层修改为ResNet残差模块， 添加改进后的空洞 空间金字塔池化模块融合不同尺度的信息， 并对编码部分添加跳跃连接， 缓解上采样的信 息丢失； [0007]步骤二、 对训练样本进行 数据增强， 归一 化处理； [0008]步骤三、 将处理后的图像数据引入训练好的改进Unet模型， 得到分割后的结果， 最 终获取遥感中的道路信息 。 [0009]进一步的， 步骤一中， 所述改进后的空洞 空间金字塔池化模块介于解码网络与编 码网络之间添加， 所述跳跃 连接在解码网络中添加。 [0010]进一步的， 步骤二和步骤三中， 具体步骤 包括： [0011]S1、 采集遥感图像， 并对遥感图像进行图像预处 理； [0012]S2、 对预处理后的图像进行标注， 将标注后的图像调整为统一尺寸， 制作训练数据 集； [0013]S3、 将得到的训练数据集作为改进Unet模型的输入图像数据， 训练改进Unet分割 模型；说　明　书 1/3 页 3 CN 115512220 A 3