Image classification: ValueError: too many values to unpack (expected 2)

Question

我正在练习图像分类项目。我得到

这是我使用的代码。任何人都可以建议我做错了什么。

# Predict
test_batch_x, test_batch_y = test_gen.next()
pred_batch = model.predict(test_batch_x)

test_labels = np.argmax(test_batch_y, axis=1)
test_pred = np.argmax(pred_batch, axis=1)

它在 test_batch_x, test_batch_y = test_gen.next()

处抛出此错误

 ValueError: too many values to unpack (expected 2)

有人可以对此提出建议吗？

Answer 1

看来，方法 next() 没有 return 两个值， 但您尝试将结果解压缩为两个变量。

由于您使用 flow_from_directory 创建了 test_gen，它是一个 tf.data.Dataset：https://www.tensorflow.org/api_docs/python/tf/data/Dataset。

我认为下一个只有 return 一个值，因为没有定义标签 class_mode=None。