代码详解:构建一个简单的Keras+深度学习REST API( 六 )

如果你正在使用自定义模型 , 则需要修改此函数以从磁盘加载架构+权重 。

在对任何来自客户端的数据进行预测之前 , 首先需要准备并预处理数据:

def prepare_image(image target):

# if the image mode is not RGB convert it

if image.mode != "RGB":

image = image.convert("RGB")

# resize the input image and preprocess it

image = image.resize(target)

image = img_to_array(image)

image = np.expand_dims(image axis=0)

image = imagenet_utils.preprocess_input(image)

# return the processed image

推荐阅读