【毕业设计】基于 U-Net 网络的遥感图像语义分割

文件列表(压缩包大小 46.93M)      点击页面右上角“下载”按钮下载

概述

基于 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 未来展望 致谢 参考文献

来源https://github.com/gzr2017/UNet-AerialImageSegmentation

理工酷提示:

如果遇到文件不能下载或其他产品问题,请添加管理员微信:ligongku666,并备注:产品反馈

评论(1)

0/250
阿托 • 6917
数据集来源:https://project.inria.fr/aerialimagelabeling/download/
2021-01-28 回复
免费
赞 6
收藏 0
评论 1
举报

网站声明:本站所有资源均为用户上传,如果侵犯了您的合法权益,请点击上方举报按钮,或添加管理员微信:ligongku666 ,并备注:举报。我们将快速核实并处理。

文件编号:562
上传时间:2021-01-27
文件大小:46.93M

阿托

声望 • 6917

分类:
计算机/软件
标签:
机器学习毕业设计