SAR/split_train_val_test.py

import glob
import os
import shutil
import random
def split(src_lists="",dst_path="D:/data/data_train_val_test"):
    os.makedirs(dst_path,exist_ok=True)
    f_train=open(dst_path+"/train.txt","w",encoding="utf-8")
    f_val=open(dst_path+"/val.txt","w",encoding="utf-8")
    f_test=open(dst_path+"/test.txt","w",encoding="utf-8")
    random.seed(111)
    random.shuffle(src_lists)

    l=len(src_lists)
    val_i=int(l*0.7)
    test_i=int(l*1.0)
    #print(val_i,test_i,l)
    for file in src_lists[:val_i]:
        f_train.writelines(file+"\n")
    for file in src_lists[val_i:test_i]:
        f_val.writelines(file+"\n")
    for file in src_lists[test_i:]:
        f_test.writelines(file+"\n")
    f_train.close()
    f_val.close()
    f_test.close()