SageMaker 部署自定义脚本



我是SageMaker的新手,所以如果我错过了一些明显的东西,我很抱歉。

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

据我了解,我可以使用 sagemaker 部署我的模型,并对来自 Lambda 的每一帧进行预测,不幸的是,这不是一个选择,因为 ffprobe、ffmpeg 和 numpy 太大而无法适应有限的 lambda 空间。

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

Sagemaker 允许您使用自定义 Docker 镜像(AWS 文档(

构建

您自己的自定义容器映像:如果没有预构建的亚马逊 可用于或修改高级的 SageMaker 容器映像 方案中,您可以打包自己的脚本或算法以用于 Amazon SageMaker.您可以使用任何编程语言或框架来 开发您的容器

  • 使用您的代码(FFmpeg,TensorFlow(创建一个docker镜像
  • 在本地测试 docker 容器
  • 在 Amazon ECR(弹性容器存储库(上部署映像
  • 创建 SageMaker 模型并指向图像

有关详细信息,您可以从本教程中了解更多信息

最新更新