Sfoglia il codice sorgente

fix label list in remote sensing

FlyingQianMM 4 anni fa
parent
commit
5748846a76

+ 1 - 1
docs/examples/remote_sensing.md

@@ -24,7 +24,7 @@ cd PaddleX/examples/remote_sensing/
 
 ## 数据准备
 
-本案例使用2015 CCF大数据比赛提供的高清遥感影像,包含5张带标注的RGB图像,图像尺寸最大有7969 × 7939、最小有4011 × 2470。该数据集共标注了5类物体,分别是背景(标记为0)、植被(标记为1)、建筑(标记为2)、水体(标记为3)、道路 (标记为4)。
+本案例使用2015 CCF大数据比赛提供的高清遥感影像,包含5张带标注的RGB图像,图像尺寸最大有7969 × 7939、最小有4011 × 2470。该数据集共标注了5类物体,分别是背景(标记为0)、植被(标记为1)、道路(标记为2)、建筑(标记为3)、水体(标记为4)。
 
 本案例将前4张图片划分入训练集,第5张图片作为验证集。为增加训练时的批量大小,以滑动窗口为(1024,1024)、步长为(512, 512)对前4张图片进行切分,加上原本的4张大尺寸图片,训练集一共有688张图片。在训练过程中直接对大图片进行验证会导致显存不足,为避免此类问题的出现,针对验证集以滑动窗口为(769, 769)、步长为(769,769)对第5张图片进行切分,得到40张子图片。
 

+ 1 - 1
examples/remote_sensing/README.md

@@ -30,7 +30,7 @@ cd PaddleX/examples/remote_sensing/
 
 ## <h2 id="1">数据准备</h2>
 
-本案例使用2015 CCF大数据比赛提供的高清遥感影像,包含5张带标注的RGB图像,图像尺寸最大有7969 × 7939、最小有4011 × 2470。该数据集共标注了5类物体,分别是背景(标记为0)、植被(标记为1)、建筑(标记为2)、水体(标记为3)、道路 (标记为4)。
+本案例使用2015 CCF大数据比赛提供的高清遥感影像,包含5张带标注的RGB图像,图像尺寸最大有7969 × 7939、最小有4011 × 2470。该数据集共标注了5类物体,分别是背景(标记为0)、植被(标记为1)、道路(标记为2)、建筑(标记为3)、水体 (标记为4)。
 
 本案例将前4张图片划分入训练集,第5张图片作为验证集。为增加训练时的批量大小,以滑动窗口为(1024,1024)、步长为(512, 512)对前4张图片进行切分,加上原本的4张大尺寸图片,训练集一共有688张图片。在训练过程中直接对大图片进行验证会导致显存不足,为避免此类问题的出现,针对验证集以滑动窗口为(769, 769)、步长为(769,769)对第5张图片进行切分,得到40张子图片。
 

+ 1 - 1
examples/remote_sensing/prepara_data.py

@@ -86,7 +86,7 @@ for h in range(0, H, val_stride[1]):
         val_tile_id += 1
 
 # 生成labels.txt
-label_list = ['background', 'vegetation', 'building', 'water', 'road']
+label_list = ['background', 'vegetation', 'road', 'building', 'water']
 for i, label in enumerate(label_list):
     mode = 'w' if i == 0 else 'a'
     with open('./dataset/labels.txt', 'a') as f: