Fix train alpha list in binary to str in linux os

SoongE · Jan 30, 2021 · 9920fc6 · 9920fc6
1 parent 24c6653
commit 9920fc6
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/data_prepare.py b/data_prepare.py
@@ -22,11 +22,7 @@ def prepare_data():
     background_dir = "data/unzip/background"
     evaluation_dir = "data/unzip/evaluation"
     processed_dir = "data/processed"
-
-    train_dir = os.path.join(processed_dir, 'train')
-    val_dir = os.path.join(processed_dir, 'val')
-    test_dir = os.path.join(processed_dir, 'test')
-
+
     random.seed(5)
 
     if not os.path.exists(processed_dir):
@@ -45,6 +41,7 @@ def prepare_data():
 
     # Split background data into train, validation data and make test data
     train_alpha = list(np.random.choice(back_alpha, size=30, replace=False))
+    train_alpha = [str(x) for x in train_alpha]
     val_alpha = [x for x in back_alpha if x not in train_alpha]
     test_alpha = [x for x in glob(evaluation_dir + '/*')]
     test_alpha.sort()
@@ -53,10 +50,6 @@ def prepare_data():
     val_dir = os.path.join(processed_dir, 'val')
     test_dir = os.path.join(processed_dir, 'test')
 
-    print(len(train_alpha))
-    print(len(val_alpha))
-    print(len(test_alpha))
-
     move_image_to_processed_dir(train_alpha, train_dir, 'train')
     move_image_to_processed_dir(val_alpha, val_dir, 'val')
     move_image_to_processed_dir(test_alpha, test_dir, 'test')