Add random crops while training

titu1994 · Jan 5, 2018 · 6fba4d2 · 6fba4d2
1 parent 1177b4f
commit 6fba4d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/data_loader.py b/data_loader.py
@@ -53,7 +53,8 @@
 def parse_data(filename, scores):
     image = tf.read_file(filename)
     image = tf.image.decode_jpeg(image, channels=3)
-    image = tf.image.resize_images(image, (IMAGE_SIZE, IMAGE_SIZE))
+    image = tf.image.resize_images(image, (256, 256))
+    image = tf.random_crop(image, size=(IMAGE_SIZE, IMAGE_SIZE, 3))
     image = tf.image.random_flip_left_right(image)
     image = (tf.cast(image, tf.float32) - 127.5) / 127.5
     return image, scores

diff --git a/weights/mobilenet_weights.h5 b/weights/mobilenet_weights.h5