将预处理步骤捆绑到 Tensorflow SavedModel

Question

我已经使用 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

Answer 1

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

例如inferenceFile.py imageToInfer.png

一个开箱即用的想法是编写一个不同的深度学习模型，作为输入您的非预处理图像并将您提供给模型的预处理图像输出，不确定这是否可以实现。