Преглед на файлове

Fix bug caused by signature change of resize_images().

Christopher Shallue преди 9 години
родител
ревизия
1e16f101f8
променени са 1 файла, в които са добавени 1 реда и са изтрити 2 реда
  1. 1 2
      im2txt/im2txt/ops/image_processing.py

+ 1 - 2
im2txt/im2txt/ops/image_processing.py

@@ -109,8 +109,7 @@ def process_image(encoded_image,
   assert (resize_height > 0) == (resize_width > 0)
   if resize_height:
     image = tf.image.resize_images(image,
-                                   new_height=resize_height,
-                                   new_width=resize_width,
+                                   size=[resize_height, resize_width],
                                    method=tf.image.ResizeMethod.BILINEAR)
 
   # Crop to final dimensions.