此软件仓库包含针对不同遥感数据集的不同语义分割模型的 PyTorch 实现 👨💻👨💻👨💻
This repository contains PyTorch implementations of different semantic segmentation models for different remote sensing datasets
Loading..........
数据集为城市区域的遥感影像语义分割数据集,特征影像数据分辨率为8米,包括38幅遥感影像,波段为红色、绿色,近红外波段,DSM波段,其中24幅完全标注,标注的土地覆盖类别包括:耕地:[0, 255, 255],林地:[255, 0, 0],草地:[0, 255, 0],建筑用地:[255, 0, 255],水域:[0, 0, 255],未利用地及其他:[255, 255, 0])。数据详情:http://www2.isprs.org/commissions/comm3/wg4/2d-sem-label-potsdam.html
Impervious surface | Building | Lowveg | Tree | Car | mIoU(%) | MeanF1(%) | OA(%) |
---|---|---|---|---|---|---|---|
96 | 98.5 | 93.9 | 91.5 | 95.6 | 89.72 | 95.1 | 94.75 |
文件夹例子
Potsdam
│ ├── train_images (original)
│ ├── train_masks (original)
│ ├── test_images (original)
│ ├── test_masks (original)
│ ├── test_masks_eroded (original)
│ ├── train (processed)
│ ├── test (processed)
因为Potsdam数据集有38块,可以选择块来做为训练集,剩下的做为测试集 也可以使用我切好的 https://www.123pan.com/s/YnBgjv-D8j1H.html提取码:pots
https://drive.google.com/file/d/1NWLLVlUVaIZYwePbPx7Ca60AdyW-cdMT/view?usp=sharing
python pytorch_remote_sensing_senmentation/tools/potsdam_cut.py --img-dir "data/potsdam/train_images" --mask-dir "data/potsdam/train_masks" --output-img-dir "data/potsdam/train/images_1024" --output-mask-dir "data/potsdam/train/masks_1024" --mode "train" --split-size 1024 --stride 512 --rgb-image
数据集包含33幅不同大小的遥感图像,每幅图像都是从一个更大的顶层正射影像图片提取的,图像选择的过程避免了出现没有数据的情况。顶层影像和DSM的空间分辨率为9 cm。遥感图像格式为8位TIFF文件,由近红外、红色和绿色3个波段组成。DSM是单波段的TIFF文件,灰度等级(对应于DSM高度)为32位浮点值编码。
切割好的数据集: 123盘:https://www.123pan.com/s/YnBgjv-rST4H.html提取码:Vsb8
Google Dirver:https://drive.google.com/file/d/1L2rxAzMm-pEV2dY111285t04hOm2L1hv/view?usp=drive_link
Impervious surface | Building | Lowveg | Tree | Car | mIoU(%) | MeanF1(%) |
---|---|---|---|---|---|---|
94.8 | 93.7 | 86.6 | 91.85 | 89.83 | 85.9 | 92.4 |
类别标签:背景 - 1、建筑 - 2、道路 - 3、水 - 4、贫瘠 - 5、森林 - 6、农业 - 7。无数据区域被指定为 0,应忽略。 南京、常州、武汉5987张高空间分辨率(0.3 m)遥感影像 关注城乡不同地理环境 推进语义分割和领域适应任务 三个巨大的挑战: 多尺度物体 复杂背景样本 类别分布不一致
数据集:LoveDA数据集在Zenodo
Background | Building | Road | Water | Barren | Forest | Agriculture | mIoU(%) | MeanF1(%) | OA(%) |
---|---|---|---|---|---|---|---|---|---|
46.85 | 61.43 | 59.07 | 82.31 | 19.94 | 49.89 | 66.15 | 55.09 | 69.19 | 70.5 |
OpenEarthMap 是全球高分辨率土地覆被制图的基准数据集。OpenEarthMap 包含 5000 幅航空和卫星图像,其中有人工标注的 8 类土地覆被标签和 220 万个片段,地面采样距离为 0.25-0.5 米,覆盖 6 大洲 44 个国家的 97 个地区。在 OpenEarthMap 上训练的土地覆被测绘模型可在全球范围内通用,并可在各种应用中作为现成模型使用。
数据集:OpenEarthMap
Background | Bareland | Rangeland | Developed | Road | Tree | Water | Agriculture | Building | mIoU |
---|---|---|---|---|---|---|---|---|---|
96.2 | 44.3 | 58.2 | 56.8 | 63.1 | 71 | 83.2 | 77.5 | 81.5 | 70.2 |
多模态多任务 VQA 数据集(EarthVQA)旨在连接地球视觉和地球科学语言,其中包括共同注册的遥感图像、土地覆盖语义掩码和任务驱动的语言文本。
EarthVQA 数据集包含 6000 幅 0.3 米的图像和 208593 个嵌入了城市和农村治理要求的 QA 对。这些 QA 对设计用于交通状况、教育设施、绿色生态、耕地状况等多种推理任务中的判断、计数、对象状况分析和综合分析类型。这种多模式、多任务的数据集提出了新的挑战,需要对遥感图像进行地理空间关系推理和归纳。
下载数据