22 lines
480 B
Python
22 lines
480 B
Python
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')
|