5

我正在使用带有 keras 的 VGG16 进行迁移学习（我的新模型中有 7 个类），因此我想使用内置的 decode_predictions 方法来输出我的模型的预测。但是，使用以下代码：

preds = model.predict(img)

decode_predictions(preds, top=3)[0]

我收到以下错误消息：

ValueError：decode_predictions需要一批预测（即形状的二维数组（样本，1000））。找到具有形状的数组：(1, 7)

现在我想知道为什么当我的再训练模型中只有 7 个类时它期望 1000 个。

我在 stackoverflow 上发现的一个类似问题（Keras: ValueError: decode_predictions 需要一批预测 ）建议在模型定义中包含 'inlcude_top=True' 来解决这个问题：

model = VGG16(weights='imagenet', include_top=True)

我已经尝试过了，但它仍然无法正常工作 - 给了我和以前一样的错误。非常感谢有关如何解决此问题的任何提示或建议。

4

2 回答 2

10

我怀疑您正在使用一些预先训练的模型，例如 resnet50 并且您正在decode_predictions像这样导入：

from keras.applications.resnet50 import decode_predictions

decode_predictions 将 (num_samples, 1000) 个概率数组转换为原始 imagenet 类的类名。

如果你想在 7 个不同的类之间进行迁移学习和分类，你需要这样做：

base_model = resnet50 (weights='imagenet', include_top=False)

# add a global spatial average pooling layer
x = base_model.output
x = GlobalAveragePooling2D()(x)
# add a fully-connected layer
x = Dense(1024, activation='relu')(x)
# and a logistic layer -- let's say we have 7 classes
predictions = Dense(7, activation='softmax')(x) 
model = Model(inputs=base_model.input, outputs=predictions)
...

拟合模型并计算预测后，您必须手动将类名分配给输出编号，而不使用导入的decode_predictions

于 2018-03-13T15:09:52.500 回答
0

'decode_predictions'函数的重载。注释掉原函数的1000个类约束：

CLASS_INDEX = None
@keras_modules_injection
def test_my_decode_predictions(*args, **kwargs):
    return my_decode_predictions(*args, **kwargs)


def my_decode_predictions(preds, top=5, **kwargs):
    global CLASS_INDEX

    backend, _, _, keras_utils = get_submodules_from_kwargs(kwargs)

    # if len(preds.shape) != 2 or preds.shape[1] != 1000:
    #     raise ValueError('`decode_predictions` expects '
    #                      'a batch of predictions '
    #                      '(i.e. a 2D array of shape (samples, 1000)). '
    #                      'Found array with shape: ' + str(preds.shape))
    if CLASS_INDEX is None:
        fpath = keras_utils.get_file(
            'imagenet_class_index.json',
            CLASS_INDEX_PATH,
            cache_subdir='models',
            file_hash='c2c37ea517e94d9795004a39431a14cb')
        with open(fpath) as f:
            CLASS_INDEX = json.load(f)
    results = []
    for pred in preds:
        top_indices = pred.argsort()[-top:][::-1]
        result = [tuple(CLASS_INDEX[str(i)]) + (pred[i],) for i in top_indices]
        result.sort(key=lambda x: x[2], reverse=True)
        results.append(result)
    return results


print('Predicted: ', test_my_decode_predictions(pred, top=10))
于 2020-03-26T10:40:03.703 回答