Amazon SageMaker 不支持的媒体类型：图像/jpeg

Question

我使用 Tensorflow 和 Keras 训练了自己的 model 进行图像分类，我正在尝试将其与亚马逊的 SageMaker 一起部署和使用。 我完成了将mymodel.h5文件转换为mymodel.tar.gz文件并将其移动到 SageMaker S3 存储桶的过程。 然后，按照教程，我使用以下方法创建了 SageMaker model：

from sagemaker.tensorflow.model import TensorFlowModel
sagemaker_model = TensorFlowModel(model_data = 's3://' + sagemaker_session.default_bucket() + '/model/model.tar.gz',
                                  role = role,
                                  framework_version = '1.12',
                                  entry_point = 'train.py')

并创建端点以访问 model：

predictor = sagemaker_model.deploy(initial_instance_count=1,
                                   instance_type='ml.m4.xlarge')

现在，由于创建此 model 的目的是为了图像分类，我正在尝试将图像传递到我的端点，但已经Received client error (415) from primary with message "{"error": "Unsupported Media Type: image/jpeg"}" 。 读完之后，我觉得我可能需要做更多的工作才能访问image/jpeg内容类型，因为默认值似乎是application/json 、 text/libsvm和text/csv 。

Answer 1

假设您的 model 已转换为 TF 服务格式，您可以使用saved_model_cli确认您的 model 期望的输入。 请参阅描述如何使用saved_model_cli的示例。 请注意如何通过将图像转换为张量来进行预测。