train

trainVal.py

@@ -0,0 +1,24 @@
+import random
+import shutil
+import os
+def trainVal(path):
+    Totaldata = []
+    for dir in os.listdir(path):
+        if dir == 'images':
+            for fileName in os.listdir(os.sep.join([path, dir])):
+                Totaldata.append(fileName)
+    imagePath = path+'/'+'images'
+    random.shuffle(Totaldata)
+    val = Totaldata[:int(len(Totaldata)/10)]
+    train = Totaldata[int(len(Totaldata)/10):]
+    with open(path+'/'+'train.txt','w') as train_:
+        for data in train:
+            train_.write(imagePath+'/'+data)
+    with open(path+'/'+'val.txt','w') as val_:
+        for data in val:
+            val_.write(imagePath+'/'+data)
+    # return train, val
+
+if __name__ == '__main__':
+    path = '/home/data/New_project/totaldata/shops_185'
+    trainVal(path)