数据分析

data_preprocessing/sub_data_preprocessing.py

@@ -1,14 +1,18 @@
 from create_extra import split_subdirs
 from data_split import split_directory
 from extend import ImageExtendProcessor
+from combine_sub_class import combine_dirs
 import yaml
 
 
 def data_preprocessing(conf):
-    # split_subdirs(conf)
-    # image_ex = ImageExtendProcessor(conf)
-    # image_ex.control_number()
-    split_directory(conf)
+    if conf['control']['split']:
+        split_subdirs(conf)
+        image_ex = ImageExtendProcessor(conf)
+        image_ex.control_number()
+        split_directory(conf)
+    if conf['control']['combine']:
+        combine_dirs(conf)
 
 
 if __name__ == '__main__':