我已经使用Tensorflow Keras API创建了三种不同的Convnets模型。客户端将图像发布到我的API网关URL，lambda收到图像，我执行预处理步骤，然后调用第一个端点并固定图像方向，然后调用第二个模型以获取一些预测，最后如果第二个模型预测一个类，我将调用另一个端点。所有这些逻辑都在我的lambda函数中，并且我已经传递了三个端点名称作为环境变量。现在，我还知道如何将所有模型放在一个端点中，并通过带有CustomAttributes参数的boto3，invoke_endpoint方法在一个端点上仅调用模型。
我的问题是：是否有一种方法可以使Tensorflow模型包含（序列化）所有这些逻辑（预处理-加载TF模型并获取预测）并返回json输出？这样我的lambda会小得多。我只需要为“ mTF模型”创建一个端点，然后调用它即可。