dataset_utils bug for tf.image.resize_images

kv91 · Oct 29, 2016 · 0fab4b1 · 0fab4b1
1 parent 5511b31
commit 0fab4b1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/session-3/libs/dataset_utils.py b/session-3/libs/dataset_utils.py
@@ -74,7 +74,7 @@ def create_input_pipeline(files, batch_size, n_epochs, shape, crop_shape=None,
     else:
         rsz_shape = [int(crop_shape[0] / crop_factor),
                      int(shape[1] / shape[0] * crop_shape[1] / crop_factor)]
-    rszs = tf.image.resize_images(imgs, rsz_shape[0], rsz_shape[1])
+    rszs = tf.image.resize_images(imgs, rsz_shape)
     crops = (tf.image.resize_image_with_crop_or_pad(
         rszs, crop_shape[0], crop_shape[1])
         if crop_shape is not None

diff --git a/session-4/libs/dataset_utils.py b/session-4/libs/dataset_utils.py
@@ -74,7 +74,7 @@ def create_input_pipeline(files, batch_size, n_epochs, shape, crop_shape=None,
     else:
         rsz_shape = [int(crop_shape[0] / crop_factor),
                      int(shape[1] / shape[0] * crop_shape[1] / crop_factor)]
-    rszs = tf.image.resize_images(imgs, rsz_shape[0], rsz_shape[1])
+    rszs = tf.image.resize_images(imgs, rsz_shape)
     crops = (tf.image.resize_image_with_crop_or_pad(
         rszs, crop_shape[0], crop_shape[1])
         if crop_shape is not None

diff --git a/session-5/libs/dataset_utils.py b/session-5/libs/dataset_utils.py
@@ -74,7 +74,7 @@ def create_input_pipeline(files, batch_size, n_epochs, shape, crop_shape=None,
     else:
         rsz_shape = [int(crop_shape[0] / crop_factor),
                      int(shape[1] / shape[0] * crop_shape[1] / crop_factor)]
-    rszs = tf.image.resize_images(imgs, rsz_shape[0], rsz_shape[1])
+    rszs = tf.image.resize_images(imgs, rsz_shape)
     crops = (tf.image.resize_image_with_crop_or_pad(
         rszs, crop_shape[0], crop_shape[1])
         if crop_shape is not None