AWSLambda调用用于语义分割的模型时,返回的是application/x-recordio-protobuf格式而不是图像。
创始人
2024-09-26 03:00:23
0

该问题通常发生在将语义分割模型部署在 AWS SageMaker 上并通过 AWS Lambda 做预测时。默认情况下,SageMaker 输出的是 application/x-recordio-protobuf 格式的数据,需要进一步处理才能得到可用的图像数据。以下是解决方法的代码示例:

import boto3
import numpy as np
import cv2
from io import BytesIO

sagemaker_runtime = boto3.client('sagemaker-runtime')

def handler(event, context):
    # 从事件中获取图像数据
    image_bytes = event['image']
    # 调用 SageMaker 服务进行预测,返回 application/x-recordio-protobuf 格式的数据
    response = sagemaker_runtime.invoke_endpoint(EndpointName='semantic-segmentation-model', 
                                                  ContentType='application/x-image', 
                                                  Body=image_bytes)
    # 解析 application/x-recordio-protobuf 格式的数据
    seg_record = response['Body'].read()
    seg_array = np.frombuffer(seg_record, dtype=np.uint8)
    # 处理 seg_array 获得分割结果
    seg_image = # do some processing on seg_array
    # 返回图像数据
    return {'segmentation': seg_image}

在上述代码中,我们通过调用 SageMaker 的客户端来从 Lambda 函数中调用模型。使用 ContentType='application/x-image' 来指定输入类型,这样我们就可以上传图像并将其转为 application/x-recordio-protobuf 格式来进行预测。然后,我们使用 numpy.frombuffer 从 response 中读取 application/x-recordio-protobuf 格式的数据,并使用 cv2 进行处理以获得图像分割结果。最后,我们可以将图像分割结果作为响应返回给调用方。

相关内容

热门资讯

6分钟辅助!hhpoker是真... 6分钟辅助!hhpoker是真的假的,hhpoker真的有透视吗,演示教程(真是有挂)1、每一步都需...
第七分钟辅助!we poker... 第七分钟辅助!we poker插件,we poker免费辅助器,手筋教程(有挂方式)暗藏猫腻,小编详...
七分钟辅助!aa poker辅... 七分钟辅助!aa poker辅助包,pokemmo脚本辅助器下载,讲义教程(有挂秘籍)1、pokem...
第四分钟辅助!wepoker辅... 第四分钟辅助!wepoker辅助器安装包定制,aapoker透视脚本,大纲教程(有挂总结)所有人都在...
第四分钟辅助!wpk透视辅助靠... 第四分钟辅助!wpk透视辅助靠谱吗,wepoker透视苹果系统,诀窍教程(有挂猫腻);运wepoke...
七分钟辅助!aapoker透视... 七分钟辅助!aapoker透视脚本入口,wejoker开挂,学习教程(有挂教程)1、打开软件启动之后...
第四分钟辅助!淘宝买wepok... 第四分钟辅助!淘宝买wepoker透视有用吗,wpk俱乐部怎么作弊,积累教程(有挂规律)1、下载好淘...
6分钟辅助!we-poker软... 6分钟辅助!we-poker软件,购买的wpk辅助在哪里下载,积累教程(有挂技巧)购买的wpk辅助在...
第2分钟辅助!如何下载wepo... 第2分钟辅助!如何下载wepoker安装包,newpoker怎么安装脚本,教程书教程(今日头条)1、...
4分钟辅助!wepoker钻石... 4分钟辅助!wepoker钻石怎么看底牌,wepoker辅助器有哪些功能,法门教程(有挂详细)该软件...