python-2.7 - Keras:Vgg16 -- `decode_predictions' 中的错误

Question

我正在尝试使用 Keras 中预训练的 VGG16 模型执行图像分类任务。我按照 Keras application page 中的说明编写的代码， 是:

from keras.applications.vgg16 import VGG16
from keras.preprocessing import image
from keras.applications.vgg16 import preprocess_input, decode_predictions
import numpy as np

model = VGG16(weights='imagenet', include_top=True)
img_path = './train/cat.1.jpg'
img = image.load_img(img_path, target_size=(224, 224))
x = image.img_to_array(img)
x = np.expand_dims(x, axis=0)
x = preprocess_input(x)

features = model.predict(x)
(inID, label) = decode_predictions(features)[0]

这与 this question 中显示的代码非常相似论坛里已经问过了。但是，尽管 include_top 参数为 True，但我收到以下错误:

Traceback (most recent call last):
  File "vgg16-keras-classifier.py", line 14, in <module>
    (inID, label) = decode_predictions(features)[0]
ValueError: too many values to unpack

任何帮助将不胜感激!谢谢!

Answer 1

这是因为(根据可能找到的函数定义 here )一个函数 decode_predictions返回一个三元组 (class_name, class_description, score) .这就是为什么它声称要解压缩的值太多。