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')