写python代码案例_keras topN显示,自编写代码案例

对于使用已经训练好的模型，比如VGG,RESNET等，keras都自带了一个keras.applications.imagenet_utils.decode_predictions的方法,有很多限制：

def decode_predictions(preds, top=5):

"""Decodes the prediction of an ImageNet model.

# Arguments

preds: Numpy tensor encoding a batch of predictions.

top: Integer, how many top-guesses to return.

# Returns

A list of lists of top class prediction tuples

`(class_name, class_description, score)`.

One list of tuples per sample in batch input.

# Raises

ValueError: In case of invalid shape of the `pred` array

(must be 2D).

"""

global CLASS_INDEX

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 (s