keras_imagenet_resnet50.py

"""Serve Keras ResNet50 model trained on ImageNet.

Prediction endpoint, served at `/predictions` takes a URL pointing to an image
and returns a list of class probabilities.
"""
from serveit.server import ModelServer
from serveit.utils import get_bytes_to_image_callback

from keras.applications.resnet50 import ResNet50
from keras.applications.resnet50 import decode_predictions
from keras.applications.resnet50 import preprocess_input

from flask import request
import requests

# load Resnet50 model pretrained on ImageNet
model = ResNet50(weights='imagenet')


# define a loader callback for the API to fetch the relevant data and
# convert to a format expected by the prediction function
def loader():
    """Load image from URL, and preprocess for Resnet."""
    url = request.args.get('url')  # read image URL as a request URL param
    response = requests.get(url)  # make request to static image file
    return response.content

# get a bytes-to-image callback, resizing the image to 224x224 for ImageNet
bytes_to_image = get_bytes_to_image_callback(image_dims=(224, 224))

# deploy model to a ModelServer
server = ModelServer(
    model,
    model.predict,
    data_loader=loader,
    preprocessor=[bytes_to_image, preprocess_input],
    postprocessor=decode_predictions,
)

# start API
server.serve()