소스 검색

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.