import os
import random


random.seed(42)

geotiffs = os.listdir(r'E:\RSdata\wlk_right_448\dataset_5m_jpg')
num = len(geotiffs)
split_rate = 0.2

eval_index = random.sample(geotiffs, k=int(num*split_rate))

f_train = open(r'E:\RSdata\wlk_right_448/train.txt', 'w')
f_val = open(r'E:\RSdata\wlk_right_448/val.txt', 'w')

# 写入文件
for geotiff in geotiffs:
    if geotiff in eval_index:
        f_train.write(str(geotiff)+'\n')
    else:
        f_val.write(str(geotiff)+'\n')