Diff of /dataloaders/split_data.py [000000] .. [903821]

Switch to side-by-side view

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