基于 U-Net 网络的遥感图像语义分割 完整代码+论文

一、研究目的

U-Net 是一种由全卷积神经网络启发的对称结构网络，在医疗影像分割领域取得了很好的效果。 此次研究尝试使用 U-Net 网络在对多光谱遥感影像数据集上进行训练，尝试使用卷积神经网络自动分割出建筑，希望能够得到一种自动分割遥感影像的简便方法。

二、研究方法

首先提出了一种基于遥感图像类别比率的交叉熵损失函数——类别平衡交叉熵。并与应用于医疗图像分割的 U-Net 相结合，将其应用于遥感图像语义分割。

在 Inria Aerial Image Labeling Dataset 训练数据集上分别使用交叉熵损失函数和类别平衡交叉熵损失函数进行训练，得到两个训练好的卷积神经网络。再利用这两个网络在 Inria Aerial Image Labeling Dataset 测试数据集上生成预测图像进行比对。

三、研究结论

两种方法在正确率和交叉熵上没有太大差别，以交叉熵作为损失函数略优于以类别平衡交叉熵方法。但是这两种方法在F1 Score 上有较大差别。交叉熵的 F1 Score 为 0.47，类别平衡交叉熵的F1 Score 为 0.51，类别平衡交叉熵较交叉熵提高了 8.5%。

四、目录

第一章 绪论

1.1 研究背景与意义

1.2 国内外研究现状

1.2.1 语义分割研究现状

1.2.2 将深度学习应用于遥感图像分割现状

1.3 本文的主要工作

1.4 论文章节安排

第二章 背景知识

2.1 全卷积网络

2.2 使用全连接网络进行精准分割

2.2.1 线性结构网络

2.2.2 对称结构网络

第三章 实验设计

3.1 数据集选择及处理

3.2 图像处理流程设计

3.2.1 网络结构

3.2.2 卷积核初始化方案

3.2.3 输出图像恢复及优化

3.3 损失设计

3.3.1 交叉熵

3.3.2 带权重的交叉熵

3.3.3 类别平衡交叉熵

3.4 结果评估

3.5 具体实现

3.5.1 实验平台

3.5.2 模型实现

第四章 实验

4.1 网络初始化设计

4.2 实验结果

4.2.1 第一组：交叉熵

4.2.2 第二组：类别平衡交叉熵

4.2.3 结果分析

4.3 典型错误

4.3.1 建筑过大

4.3.2 错认

4.3.3 树木和光影

4.4 最终结果

第五章 总结和展望

5.1 全文总结

5.2 未来展望

致谢

参考文献