From 0dab894a5be9f7d10d85e89dea91d02c71bae84d Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Sun, 16 Sep 2018 03:24:45 +0000
Subject: [PATCH] Moving files from MTGCardDetector repo
---
setup_train.py | 28 ++++++++++++++++++++++++++++
1 files changed, 28 insertions(+), 0 deletions(-)
diff --git a/setup_train.py b/setup_train.py
new file mode 100644
index 0000000..362be19
--- /dev/null
+++ b/setup_train.py
@@ -0,0 +1,28 @@
+import os
+from glob import glob
+import random
+import transform_data
+
+
+def main():
+ random.seed()
+ data_list = []
+ for subdir in glob('%s/train/*_10' % transform_data.data_dir):
+ for data in glob(subdir + "/*.jpg"):
+ data_list.append(os.path.abspath(data))
+ random.shuffle(data_list)
+
+ test_ratio = 0.1
+ test_list = data_list[:int(test_ratio * len(data_list))]
+ train_list = data_list[int(test_ratio * len(data_list)):]
+ with open('%s/train_10.txt' % transform_data.darknet_dir, 'w') as train_txt:
+ for data in train_list:
+ train_txt.write(data + '\n')
+ with open('%s/test_10.txt' % transform_data.darknet_dir, 'w') as test_txt:
+ for data in test_list:
+ test_txt.write(data + '\n')
+ return
+
+
+if __name__ == '__main__':
+ main()
--
Gitblit v1.10.0