amazon-web-services - 从 python boto 客户端在 S3 中调用带有数据负载的 SageMaker 端点

Question

我有一张图片上传到 mybucket 中的 s3。假设此数据的我的 s3 端点是 s3://mybucket/imgname

现在我还在 SageMaker 中的 sagemaker-model-endpoint 部署了一个模型

我研究了如何从 boto 客户端调用此 SageMaker 端点的示例 here但我不确定如何在 invoke_endpoint 调用中指定 s3 路径 s3://mybucket/imgname。

client = boto3.client("runtime.sagemaker", region_name = 's3.us-east-2.amazonaws.com')
client.invoke_endpoint(
      EndpointName=sagemaker-model-endpoint
      Body=payload,
      ContentType='image/jpg',
      Accept='Accept')

在这种情况下，payload 应该是什么？我在哪里指定 s3 url？

Answer 1

你需要有图片的字节，当你从S3(S3拷贝或wget)获取图片后，你可以调用:

with open(file_name, 'rb') as f:
    payload = f.read()
    payload = bytearray(payload)

client.invoke_endpoint(EndpointName=sagemaker-model-endpoint, 
                                   ContentType='application/x-image', 
                                   Body=payload)