21 lines
478 B
Python
21 lines
478 B
Python
|
import os
|
||
|
|
import random
|
||
|
|
|
||
|
|
|
||
|
|
random.seed(42)
|
||
|
|
|
||
|
|
geotiffs = os.listdir('E:\RSdata\wlk_right_224_2\dataset_5m')
|
||
|
|
num = len(geotiffs)
|
||
|
|
split_rate = 0.2
|
||
|
|
|
||
|
|
eval_index = random.sample(geotiffs, k=int(num*split_rate))
|
||
|
|
|
||
|
|
f_train = open('E:\RSdata\wlk_right_224_2/train.txt', 'w')
|
||
|
|
f_val = open('E:\RSdata\wlk_right_224_2/val.txt', 'w')
|
||
|
|
|
||
|
|
# 写入文件
|
||
|
|
for geotiff in geotiffs:
|
||
|
|
if geotiff in eval_index:
|
||
|
|
f_val.write(str(geotiff)+'\n')
|
||
|
|
else:
|
||
|
|
f_train.write(str(geotiff)+'\n')
|