将预处理步骤捆绑到Tensorflow SavedModel

时间:2019-10-07 13:57:01

标签: python tensorflow tf.keras

我已经使用Inception V3构建了一个图像分类器模型,并且已将模型保存为“ SavedModel”格式,以将其部署到生产环境中。我想知道如何将预处理步骤绑定到最终模型,以便该模型以其自然形式摄取数据。

我具有的预处理步骤是:

- resizing the image to target_size of 299, 299 using keras load_model
- change the image to numpy array
- expand dimensions
- pre_process input using inception_v3 import preprocess_input call

1 个答案:

答案 0 :(得分:0)

部署模型时,据我了解,实际部署的是用于利用模型进行推理的python代码。在此python代码中,您可以使用openCV或任何其他python库编写所有预处理的代码,并将图像作为参数传递给此python代码。

例如inferenceFile.py imageToInfer.png

一个开箱即用的想法是编写一个不同的深度学习模型,作为输入您的未预处理图像并将其输出到模型的预处理图像输出到该模型中,不确定是否可以实现。