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