feat(make_train_test): 更新数据集生成脚本

Yidadaa · Jun 6, 2019 · c2a7ebf · c2a7ebf
1 parent 66c3513
commit c2a7ebf
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 117 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,5 @@
-data/UCF
+data/UCF
+data/train
+data/test
+data/train.csv
+data/test.csv
diff --git a/.vscode/.ropeproject/config.py b/.vscode/.ropeproject/config.py
diff --git a/data/make_train_test.py b/data/make_train_test.py
@@ -11,7 +11,7 @@
 
 def split(src_dir=default_src_dir, output_dir=default_src_dir, size=default_test_size):
     # 设置默认参数
-    src_dir = default_output_dir if src_dir is None else src_dir
+    src_dir = default_src_dir if src_dir is None else src_dir
     output_dir = default_output_dir if output_dir is None else output_dir
     size = default_test_size if size is None else size
 

diff --git a/dataloader.py b/dataloader.py
@@ -18,7 +18,6 @@ def __init__(self):
             )
         ])
 
-
     def read_images(self):
         pass
-Original file line number
+Diff line change
@@ Expand Up / @@ -18,7 +18,6 @@ def __init__(self): @@
                 )
             ])
         def read_images(self):
             pass
@@ Expand Down @@