Explorar o código

Fix bug caused by signature change of resize_images().

Christopher Shallue %!s(int64=9) %!d(string=hai) anos
pai
achega
1e16f101f8
Modificáronse 1 ficheiros con 1 adicións e 2 borrados
  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.