
基于PyTorch的高分辨率遥感影像建筑物深度学习提取方法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究提出了一种利用PyTorch框架进行高分辨率遥感图像中建筑物自动识别与提取的深度学习算法,旨在提高建筑检测精度和效率。
基于DenseLinkNet50网络实现的遥感影像建筑提取方法,在效果上不逊于目前流行的Transformer模型,并且具有更高的检测效率。程序集成了训练、验证以及大图像测试的功能,其中在测试环节采用了旋转扩充数据与投票机制相结合的方法来提升识别精度。
针对遥感数据读取和处理部分,本项目基于GDAL库专门开发了tiffIO.py文件中的影像读写相关函数以提高性能。此程序曾使用SpaceNet数据集进行过训练,并支持用户采用该数据集或自行提供的其他数据集进行模型训练与测试。
具体而言,train.py负责模型的训练过程;valid.py用于验证阶段以评估模型表现;test.py则专门设计用来处理大尺寸图像的测试任务。预训练权重文件存放于weights/resnet50-19c8e357.pth中,作为基础网络的一部分被加载使用。
全部评论 (0)
还没有任何评论哟~


