From c171a6ece870be48b07d5de93ee6301b1da0c7de Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Sat, 15 Sep 2018 01:29:29 +0000
Subject: [PATCH] Trying out 10 card model setup
---
setup_train.py | 8 +++++---
1 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/setup_train.py b/setup_train.py
index d7897c9..c7a1235 100644
--- a/setup_train.py
+++ b/setup_train.py
@@ -1,11 +1,13 @@
import os
from glob import glob
import random
+import transform_data
+
def main():
random.seed()
data_list = []
- for subdir in glob('data/train/*'):
+ for subdir in glob('%s/train/*' % transform_data.data_dir):
for data in glob(subdir + "/*.jpg"):
data_list.append(os.path.abspath(data))
random.shuffle(data_list)
@@ -13,10 +15,10 @@
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('darknet/train.txt', 'w') as train_txt:
+ with open('%s/train.txt' % transform_data.darknet_dir, 'w') as train_txt:
for data in train_list:
train_txt.write(data + '\n')
- with open('darknet/test.txt', 'w') as test_txt:
+ with open('%s/test.txt' % transform_data.darknet_dir, 'w') as test_txt:
for data in test_list:
test_txt.write(data + '\n')
return
--
Gitblit v1.10.0