/ / Fehler beim Training im2txt Modell - Python, Tensorflow, Deep-Learning

Fehler beim Training im2txt-Modell - Python, Tensorflow, Deep-Learning

Ich habe versucht das zu trainieren im2txt Modell mit Tensorflow, das ich gerade von Master Branch gebaut habe,

Ich habe alle benötigten Daten heruntergeladen, aber wenn ich das Trainingsskript ausführe:

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

Es zeigt folgendes:

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"

Ist es ein Problem im Zusammenhang mit der Funktion? resize_images() oder habe ich gerade etwas falsch gemacht?

Vielen Dank

Antworten:

4 für die Antwort № 1

Aktualisieren: fix angewendet

Tut mir leid! Die Signatur der Funktion resize_images(...) In TensorFlow wurde letzte Woche geändert, was diesen Bruch verursacht hat.

Ich werde mich in Kürze darum kümmern. Wenn Sie den Fix zuvor verwenden möchten, müssen Sie die Datei im2txt / im2txt / ops / image_processing.py ändern.

Ändere einfach diese Zeile:

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

zu diesem:

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