GeoSeg 项目推荐

GeoSeg 项目推荐

【免费下载链接】GeoSeg UNetFormer: A UNet-like transformer for efficient semantic segmentation of remote sensing urban scene imagery, ISPRS. Also, including other vision transformers and CNNs for satellite, aerial image and UAV image segmentation. 项目地址: https://gitcode.com/gh_mirrors/ge/GeoSeg

1. 项目基础介绍和主要编程语言

GeoSeg 是一个基于 PyTorch、PyTorch Lightning 和 TIMM 的开源语义分割工具箱。该项目主要使用 Python 编程语言，专注于开发先进的视觉变换器（Vision Transformers）用于遥感图像的语义分割。

2. 项目核心功能

GeoSeg 的核心功能包括：

• 统一基准测试：提供了一个统一的训练脚本，支持多种分割方法的基准测试。
• 简单且有效：得益于 PyTorch Lightning 和 TIMM，代码易于进一步开发和扩展。
• 支持的遥感数据集：包括 ISPRS Vaihingen 和 Potsdam、UAVid、LoveDA 等数据集。未来将支持更多数据集。
• 多尺度训练和测试：支持对大型遥感图像进行多尺度训练和测试。
• 支持的网络模型：包括 Mamba、PyramidMamba、Vision Transformer、UNetFormer、DC-Swin、BANet、CNN、MANet、ABCNet 和 A2FPN 等。

3. 项目最近更新的功能

最近更新的功能包括：

• PyTorch 2.0 和 PyTorch Lightning 2.0 支持：更新了代码以支持 PyTorch 2.0 和 PyTorch Lightning 2.0，并支持多 GPU 训练。
• 预训练权重：提供了预训练的骨干网络权重，可以从 Google Drive 访问。
• 大型遥感图像推理：支持对大型遥感图像进行推理（inference_huge_image.py）。
• 更多网络模型：更新并提供了更多网络模型的预训练权重链接。
• 数据集处理：提供了详细的数据集预处理脚本，如 Vaihingen、Potsdam、UAVid 和 LoveDA 数据集的处理脚本。

【免费下载链接】GeoSeg UNetFormer: A UNet-like transformer for efficient semantic segmentation of remote sensing urban scene imagery, ISPRS. Also, including other vision transformers and CNNs for satellite, aerial image and UAV image segmentation. 项目地址: https://gitcode.com/gh_mirrors/ge/GeoSeg