Tensorflow ValueError：预测的批处理长度应相同

Question

我正在尝试使用TensorFlow Estimator API可视化卷积自动编码器的输出

input_fn = tf.estimator.inputs.numpy_input_fn(
    x={"x": train_data},
    y=None,
    batch_size=8,
    num_epochs=None,
    shuffle=True)

autoencoder = tf.estimator.Estimator(model_fn=autoencoder_model_fn, model_dir=model_dir)
tensors_to_log = {"loss": "loss"}
logging_hook = tf.train.LoggingTensorHook(tensors=tensors_to_log, every_n_iter=1000)

autoencoder.train(
    input_fn=input_fn,
    steps=50000,
    hooks=[logging_hook])

input_fn_predict = tf.estimator.inputs.numpy_input_fn(
    x={"x": train_data},
    y=None,
    batch_size=1,
    num_epochs=None,
    shuffle=False)
predictions = autoencoder.predict(input_fn=input_fn_predict)
predictions = [p['decoded_image'] for p in predictions]
print predictions[0].shape

我收到以下错误：

Traceback (most recent call last):
  File "AutoEncoder.py", line 164, in <module>
    main()
  File "AutoEncoder.py", line 157, in main
    predictions = [p['decoded_image'] for p in predictions]
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/estimator.py", line 425, in predict
    for i in range(self._extract_batch_length(preds_evaluated)):
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/estimator.py", line 592, in _extract_batch_length
    'different batch length then others.' % key)
ValueError: Batch length of predictions should be same. features has different batch length then others.

有人可以看到我做错了吗？ 据我了解，我在预测期间的批量大小是一个常数，等于1 ...预先感谢！

Answer 1

我成功解决了这个问题。 我刚刚更新了tensorflow （我有1.6版本，您需要1.7具有不同的批处理大小）

要查看您的tensorflow版本，您只需要在批处理中：

>python
>>>import tensorflow as tf
>>>print(tf.__version__)

然后，当您拥有1.7版本（或更高版本）时，可以在内部使用预测参数：

yield_single_examples =假

（默认情况下为True ，并且您将遇到相同的错误）。

在您的代码中，它将是：

predictions = autoencoder.predict(input_fn=input_fn_predict, yield_single_examples=False)