【人工智能】ISPRS Vaihingen & Potsdam数据集在GeoSeg框架遥感图像语义分割领域的拆分方法

下载完成数据集后解压出来，寻找下面分割方法中需要的压缩文件。明确在UNetFormer原论文中，作者所用的训练和测试图片的id。如下图所示：

Vaihingen数据集分割方式：

1. 所用的ZIP文件：
   • ISPRS_semantic_labeling_Vaihingen.zip
   • ISPRS_semantic_labeling_Vaihingen_ground_truth_COMPLETE.zip
   • ISPRS_semantic_labeling_Vaihingen_ground_truth_eroded_COMPLETE.zip
2. 数据分配：
   • train_masks：从ISPRS_semantic_labeling_Vaihingen.zip中的gts_for_participants文件夹获取。
   • train_images：从ISPRS_semantic_labeling_Vaihingen.zip中的top文件夹中，选择对应训练集（train + val）ID的文件。
   • test_masks：从ISPRS_semantic_labeling_Vaihingen_ground_truth_COMPLETE.zip中选择对应测试集ID的文件。
   • test_masks_eroded：从ISPRS_semantic_labeling_Vaihingen_ground_truth_eroded_COMPLETE.zip中选择对应测试集ID的文件。
   • test_images：从ISPRS_semantic_labeling_Vaihingen.zip中的top文件夹中，选择对应测试集ID的文件。
3. 数据切割： 使用vaihingen_patch_split.py脚本进行数据切割。

Potsdam数据集分割方式：

1. 所用的ZIP文件：
   • 2_Ortho_RGB.zip
   • 5_Labels_all.zip
   • 5_Labels_for_all_no_Boundary.zip
2. 数据分配：
   • train_images：从2_Ortho_RGB.zip中选择对应训练集（train + val）ID的文件。
   • test_images：从2_Ortho_RGB.zip中选择对应测试集ID的文件。
   • train_masks：从5_Labels_all.zip中选择对应训练集（train + val）ID的文件。
   • test_masks：从5_Labels_all.zip中选择对应测试集ID的文件。
   • test_masks_eroded：从5_Labels_for_all_no_Boundary.zip中选择对应测试集ID的文件。
3. 数据切割： 使用potsdam_patch_split.py脚本进行数据切割。

分割脚本和目录结构请查看GeoSeg框架readme：https://github.com/WangLibo1995/GeoSeg

下面我直接提供分割好的数据集，如需可直接下载。