Increase training size

satojkovic · Jan 1, 2018 · 19a3be1 · 19a3be1
1 parent 5c572e9
commit 19a3be1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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):