update

cl0ver012 · Jun 14, 2019 · f5875f7 · f5875f7
1 parent 13736aa
commit f5875f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pre_process.py b/pre_process.py
@@ -18,7 +18,7 @@
     # Path to folder where you want the composited images to go
     out_path = 'data/merged/'
 
-    train_folder = 'Combined_Dataset/Training_set/'
+    train_folder = 'data/Combined_Dataset/Training_set/'
 
     # if not os.path.exists('Combined_Dataset'):
     zip_file = 'data/Adobe_Deep_Matting_Dataset.zip'
@@ -79,7 +79,7 @@
     out_test_path = 'data/merged_test/'
 
     # test data gen
-    test_folder = 'Combined_Dataset/Test_set/'
+    test_folder = 'data/Combined_Dataset/Test_set/'
 
     if not os.path.exists(bg_test_path):
         os.makedirs(bg_test_path)