如何使用Sagemaker中经过训练的图像分类模型对通过网页上传的图像进行推理？

Question

我已经使用sagemaker训练了我的图像分类模型，并使用端点进行了部署。我还创建了一个接受图像的网页。此外，我还能够创建api。我不确定如何创建lambda函数来接受图像并对图像执行推断。另外，我是否应该在lambda函数中包含预处理步骤。

提前谢谢你。

Answer 1

是否应该包括预处理步骤可能取决于算法。对于SageMaker内置的图像分类，不需要任何预处理。在这种情况下，您可以使用Amazon API Gateway将请求转发到端点。请参阅本教程https://aws.amazon.com/blogs/machine-learning/creating-a-machine-learning-powered-rest-api-with-amazon-api-gateway-mapping-templates-and-amazon-sagemaker/。

如果不需要预处理，则不需要创建VTL模板。您可以配置"passthrough“内容处理”当没有模板与请求的Content-Type头部匹配时“。对于内置的图像分类，请确保在API网关的Method Request配置中添加Content-Type和Accept头部。