From 19a3be162493cbdcaa97726f9edb0db60b7ba482 Mon Sep 17 00:00:00 2001
From: satojkovic <satojkovic@gmail.com>
Date: Mon, 1 Jan 2018 20:53:27 +0900
Subject: [PATCH] Increase training size

---
 crop_and_aug.py         | 2 +-
 gen_train_valid_test.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/crop_and_aug.py b/crop_and_aug.py
index cf894f6..1e55c70 100644
--- a/crop_and_aug.py
+++ b/crop_and_aug.py
@@ -232,7 +232,7 @@ def do_train_test_split():
                 os.path.join(common.CROPPED_AUG_IMAGE_DIR, class_name))
         ]
         # train=0.75, test=0.25
-        train_imgs, test_imgs = train_test_split(imgs)
+        train_imgs, test_imgs = train_test_split(imgs, train_size=0.9)
         # move images to train or test directory
         os.makedirs(
             os.path.join(common.CROPPED_AUG_IMAGE_DIR, class_name, 'train'))
diff --git a/gen_train_valid_test.py b/gen_train_valid_test.py
index ae387ec..133885e 100644
--- a/gen_train_valid_test.py
+++ b/gen_train_valid_test.py
@@ -33,9 +33,9 @@
 PIXEL_DEPTH = 255.0
 PICKLE_FILENAME = 'deep_logo.pickle'
 
-TRAIN_SIZE = 50000  # prune the training data as needed. There are 163169 training files.
+TRAIN_SIZE = 70000
 VALID_SIZE = 5000
-TEST_SIZE = 5000  # There are 54425 test files.
+TEST_SIZE = 5000
 
 
 def load_logo(data_dir):