change data numbers

zz44l43 · Dec 13, 2017 · ab806a3 · ab806a3
1 parent 01c7275
commit ab806a3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/datasets/convert_quiz.py b/datasets/convert_quiz.py
@@ -36,7 +36,7 @@
 
 from datasets import dataset_utils
 
-_NUM_VALIDATION = 5000
+_NUM_VALIDATION = 10000
 
 # Seed for repeatability.
 _RANDOM_SEED = 0
@@ -52,10 +52,11 @@ def __init__(self):
         # Initializes function that decodes RGB JPEG data.
         self._decode_jpeg_data = tf.placeholder(dtype=tf.string)
         self._decode_jpeg = tf.image.decode_jpeg(self._decode_jpeg_data, channels=3)
+        self._decode_jpeg = tf.image.resize_images(self._decode_jpeg, (320, 320))
 
     def read_image_dims(self, sess, image_data):
-        image = self.decode_jpeg(sess, image_data)
-        return image.shape[0], image.shape[1]
+        #image = self.decode_jpeg(sess, image_data)
+        return 320, 320
 
     def decode_jpeg(self, sess, image_data):
         image = sess.run(self._decode_jpeg,

diff --git a/datasets/quiz.py b/datasets/quiz.py
@@ -31,9 +31,9 @@
 
 _FILE_PATTERN = 'quiz_%s_*.tfrecord'
 
-SPLITS_TO_SIZES = {'train': 45000, 'validation': 5000}
+SPLITS_TO_SIZES = {'train': 50000, 'validation': 10000}
 
-_NUM_CLASSES = 100
+_NUM_CLASSES = 200
 
 _ITEMS_TO_DESCRIPTIONS = {
     'image': 'A color image of varying size.',