Python keras.applications.resnet50.decode_predictions() Examples

The following are 1 code examples of keras.applications.resnet50.decode_predictions(). You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may also want to check out all available functions/classes of the module keras.applications.resnet50 , or try the search function .
Example #1
Source Project: Elphas   Author: scorelab   File:    License: Apache License 2.0 6 votes vote down vote up
def predict(model, img, target_size, top_n=3):
    """Run model prediction on image
        model: keras model
        img: PIL format image
        target_size: (w,h) tuple
        top_n: # of top predictions to return
            list of predicted labels and their probabilities

    if img.size != target_size:
        img = img.resize(target_size)

    x = image.img_to_array(img)
    x = np.expand_dims(x, axis=0)
    x = preprocess_input(x)

    with graph.as_default():
        preds = model.predict(x)

    return decode_predictions(preds, top=top_n)[0]