SageMaker部署自定义脚本

问题描述 投票:0回答:1

我对SageMaker很陌生,所以如果我错过明显的事情,我感到抱歉。

我已经训练了一个DL模型,该模型使用视频中的帧进行预测。在SageMaker jupyter-notebook中运行的当前脚本将视频URL作为输入,并使用FFMPEG子处理管道提取帧并随后进行预测。效果很好,但现在我想从Lambda启动该脚本。

据我所知,我可以使用sagemaker部署模型并为Lambda的每个帧进行预测,不幸的是,这不是一个选择,因为ffprobe,ffmpeg和numpy太大而无法容纳在有限的lambda空间中。 >

tl; dr:是否可以在SageMaker中将我的自定义脚本(ffmpeg帧提取+张量流模型预测)作为端点运行?

我对SageMaker来说还很陌生,所以如果我错过明显的事情,我感到抱歉。我已经训练了一个DL模型,该模型使用视频中的帧进行预测。当前脚本,在SageMaker中运行...

amazon-web-services amazon-sagemaker
1个回答
0
投票

Sagemaker允许您使用自定义Docker映像(AWS document

© www.soinside.com 2019 - 2024. All rights reserved.