From 504ece5b00f192d5c1b343fd06ce1648f9139180 Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Mon, 17 Sep 2018 03:06:19 +0000
Subject: [PATCH] Code cleaning & training new YOLO model
---
setup_train.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/setup_train.py b/setup_train.py
index c7a1235..cbbb6a7 100644
--- a/setup_train.py
+++ b/setup_train.py
@@ -7,7 +7,7 @@
def main():
random.seed()
data_list = []
- for subdir in glob('%s/train/*' % transform_data.data_dir):
+ for subdir in glob('%s/train/*_update' % transform_data.data_dir):
for data in glob(subdir + "/*.jpg"):
data_list.append(os.path.abspath(data))
random.shuffle(data_list)
--
Gitblit v1.10.0