Question

我是CNN和机器学习的新手，并且一直在尝试遵循TensorFlow的图像分类教程。

现在，可以here找到Google Colab。我遵循了TensorFlow的this official tutorial。而且我对其进行了少许更改，因此将模型保存为h5而不是tf格式，因此我可以使用Keras的model.predict_classes。

现在，我已经训练了模型，可以从保存的模型中重新加载模型。但是，每当我试图预测正在执行的图像时，我都会反复遇到list index out of range错误：

def predict():
  image = tf.io.read_file('target.jpeg')
  image = tf.image.decode_jpeg(image, channels=3)
  image = tf.image.resize(image, [224, 224])
  print(model.predict_classes(image)[0])

target.jpeg是我从训练模型的flowers_photos数据集中拍摄的图像之一。

回溯是：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 7, in predict
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/engine/sequential.py", line 319, in predict_classes
    proba = self.predict(x, batch_size=batch_size, verbose=verbose)
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/engine/training.py", line 821, in predict
    use_multiprocessing=use_multiprocessing)
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/engine/training_arrays.py", line 712, in predict
    callbacks=callbacks)
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/engine/training_arrays.py", line 187, in model_iteration
    f = _make_execution_function(model, mode)
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/engine/training_arrays.py", line 555, in _make_execution_function
    return model._make_execution_function(mode)
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/engine/training.py", line 2037, in _make_execution_function
    self._make_predict_function()
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/engine/training.py", line 2027, in _make_predict_function
    **kwargs)
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/backend.py", line 3544, in function
    return EagerExecutionFunction(inputs, outputs, updates=updates, name=name)
  File "/home/amitjoki/.local/lib/python3.6/site-packages/tensorflow/python/keras/backend.py", line 3468, in __init__
    self.outputs[0] = array_ops.identity(self.outputs[0])
IndexError: list index out of range

我进行了广泛搜索，但找不到任何解决方案。如果有人可以指出我的安装方向和运行方向，将会很有帮助。

Answer 1

Keras中的所有预测函数都需要一批输入。因此，由于要对单个图像进行预测，因此需要在图像张量的开头添加一个轴以表示批处理轴：

image = tf.expand_dims(image, axis=0)   # the shape would be (1, 224, 224, 3)
print(model.predict_classes(image)[0])

Keras Forecast_classes方法返回“列表索引超出范围”错误

1 个答案: