瀏覽代碼

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.