-
Notifications
You must be signed in to change notification settings - Fork 1
/
Fsplit.py
32 lines (24 loc) · 830 Bytes
/
Fsplit.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import numpy as np
import os
import glob
import random
import shutil
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--train_images_folder', type=str, default='/home/mariapap/DATA/SPACENET7/train/',
help='downloaded folder of SpaceNet7 training images')
args = parser.parse_args()
FOLDER = glob.glob(args.train_images_folder + '*_13*') #give your '/train/' folder destination
random.shuffle(FOLDER)
Ftrain = FOLDER[0:40]
Fval = FOLDER[40:50]
Ftest = FOLDER[50:60]
if os.path.exists('Fsplit'):
shutil.rmtree('Fsplit')
os.mkdir('Fsplit')
np.save('./Fsplit/Ftrain.npy', Ftrain)
np.save('./Fsplit/Fval.npy', Fval)
np.save('./Fsplit/Ftest.npy', Ftest)
print(len(Ftrain), 'folders for training')
print(len(Fval), 'folders for validation')
print(len(Ftest), 'folders for testing')