推理脚本，单独的负载模型和预测

Question

我有一个推理脚本，其中加载模型和预测位于一个脚本中，如何将它们分开，因此我不必每次都要进行预测时都加载模型

from keras.models import load_model

# load model
model = load_model('model.h5')

# prediction
result = model.predict(image)

Answer 1

将模型保留为全局变量，并作为该模块内部的函数进行预测。

from keras.models import load_model
model = None
# load model
def load_model_weights():
    model = load_model('model.h5')

def predict(image):
    if(model is None):
        load_model_weights()
    else:
        # prediction
        result = model.predict(image)
    return result

导入从该模块预测的功能，并在其他模块中使用。