--- a +++ b/dataloaders/split_data.py @@ -0,0 +1,11 @@ +import os +from sklearn.model_selection import train_test_split + +data_path = 'E:/data/LASet' +names = os.listdir(os.path.join(data_path,'origin')) +train_ids,test_ids = train_test_split(names,test_size=0.2,random_state=367) +with open(os.path.join(data_path,'train.list'),'w') as f: + f.write('\n'.join(train_ids)) +with open(os.path.join(data_path,'test.list'),'w') as f: + f.write('\n'.join(test_ids)) +print(len(names),len(train_ids),len(test_ids)) \ No newline at end of file