训练数据前置处理与提升训练效率

data_preprocessing/sub_data_preprocessing.py

@@ -0,0 +1,17 @@
+from create_extra import split_subdirs
+from data_split import split_directory
+from extend import ImageExtendProcessor
+import yaml
+
+
+def data_preprocessing(conf):
+    # split_subdirs(conf)
+    # image_ex = ImageExtendProcessor(conf)
+    # image_ex.control_number()
+    split_directory(conf)
+
+
+if __name__ == '__main__':
+    with open('../configs/sub_data.yml', 'r') as f:
+        conf = yaml.load(f, Loader=yaml.FullLoader)
+    data_preprocessing(conf)