update

ace19-dev · Jul 31, 2019 · db5ade9 · db5ade9
1 parent 530e517
commit db5ade9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/train.py b/train.py
@@ -339,8 +339,8 @@ def main(unused_argv):
                                            num_classes,
                                            FLAGS.how_many_training_epochs,
                                            VALIDATE_DATA_SIZE,
-                                           FLAGS.height,
-                                           FLAGS.width)
+                                           250,
+                                           250)
             val_iterator = val_dataset.dataset.make_initializable_iterator()
             val_next_batch = val_iterator.get_next()
 

diff --git a/val_data.py b/val_data.py
@@ -10,7 +10,7 @@
 MEAN=[0.485, 0.456, 0.406]
 STD=[0.229, 0.224, 0.225]
 
-RANDOM_CROP_SIZE = 200
+RANDOM_CROP_SIZE = 224
 
 
 class Dataset(object):
@@ -90,7 +90,7 @@ def tencrop(self, filename, image, label):
         images = []
         for i in range(5):
             img = tf.random_crop(image, [RANDOM_CROP_SIZE, RANDOM_CROP_SIZE, 3])
-            img = tf.image.resize(img, [self.resize_h, self.resize_w])
+            # img = tf.image.resize(img, [self.resize_h, self.resize_w])
             images.append(img)
             images.append(tf.image.flip_left_right(img))