ошибки при подготовке im2txt моделей

Question

Я пытался обучить im2txt модели с использованием Tensorflow, что я только что построенной из главной ветви,

я скачал все данные, устанавливает, что необходимо, но когда я запускаю учебный сценарий:

bazel-bin/im2txt/train \ --input_file_pattern="${MSCOCO_DIR}/train-?????-of-00256" \ --inception_checkpoint_file="${INCEPTION_CHECKPOINT}" \ --train_dir="${MODEL_DIR}/train" \ --train_inception=false \ --number_of_steps=1000000 

это показывает следующее:

Traceback (most recent call last): 
File "/home/rvl224/models/im2txt/bazel-bin/im2txt/train.runfiles/im2txt/im2txt/train.py", line 111, in tf.app.run() 
File "/home/rvl224/anaconda2/lib/python2.7/site-packages/tensorflow/python/platform/app.py", line 30, in run sys.exit(main(sys.argv[:1] + flags_passthrough)) 
File "/home/rvl224/models/im2txt/bazel-bin/im2txt/train.runfiles/im2txt/im2txt/train.py", line 65, in main model.build() 
File "/home/rvl224/models/im2txt/bazel-bin/im2txt/train.runfiles/im2txt/im2txt/show_and_tell_model.py", line 358, in build self.build_inputs() 
File "/home/rvl224/models/im2txt/bazel-bin/im2txt/train.runfiles/im2txt/im2txt/show_and_tell_model.py", line 165, in build_inputs image = self.process_image(encoded_image, thread_id=thread_id) 
File "/home/rvl224/models/im2txt/bazel-bin/im2txt/train.runfiles/im2txt/im2txt/show_and_tell_model.py", line 119, in process_image image_format=self.config.image_format) 
File "/home/rvl224/models/im2txt/bazel-bin/im2txt/train.runfiles/im2txt/im2txt/ops/image_processing.py", line 114, in process_image method=tf.image.ResizeMethod.BILINEAR) 
TypeError: resize_images() got an unexpected keyword argument 'new_height' 

ли это проблема, связанная с функцией resize_images() или я просто сделал что-то не так?

Благодаря

Answer 1

Update:fix applied

К сожалению об этом! На прошлой неделе была изменена сигнатура функции resize_images(...) в TensorFlow, что вызвало этот полом.

В ближайшее время я получу исправление. Если вы хотите использовать исправление до этого, вам необходимо изменить файл im2txt/im2txt/ops/image_processing.py.

Просто измените эту строку:

image = tf.image.resize_images(image, 
           new_height=resize_height, 
           new_width=resize_width, 
           method=tf.image.ResizeMethod.BILINEAR) 

к этому:

image = tf.image.resize_images(image, 
           size=[resize_height, resize_width], 
           method=tf.image.ResizeMethod.BILINEAR)