跳至主要内容

AWS 解决方案库

使用 Amazon SageMaker 创建个性化头像的指南

概览

本指南将介绍如何校准和部署 Stable Diffusion 模型,以通过简单的文本提示生成个性化头像。Stable Diffusion 是一种文本到图像模型,由某种利用机器学习最新进展的人工智能(AI)生成。此处所说的模型由 Amazon SageMaker 构建,并使用 DreamBooth 方法进行校准,该方法使用 10-15 张用户图像来捕捉对象的精确细节。该模型生成的个性化头像可用于社交媒体、游戏和虚拟活动等多种应用场景。该指南还涉及文本提示功能,使用户可以根据特定的文本输入生成头像。此功能扩展了应用程序的能力,使媒体和娱乐组织可以通过更多方式,来为消费者量身定制个性化内容。

本指南介绍了一种基于人工智能的方法,可帮助媒体和娱乐组织大规模开发个性化的定制内容。但是,本指南的用户应采取预防措施,确保这些人工智能功能不被滥用或操纵。访问使用 Amazon AI 内容审核服务的安全图像生成和传播模型,了解如何通过适当的审核机制保护内容。

工作原理

这些技术细节包含一张架构图,用于说明如何有效使用本解决方案。该架构图展示了关键组件及其相互作用,并逐步概述了架构的结构和功能。

Well-Architected 支柱

上面的架构图是按照 Well-Architected 最佳实践创建的解决方案示例。要做到完全的良好架构,您应该遵循尽可能多的 Well-Architected 最佳实践。

本指南中使用了 SageMaker 多模型终端节点亚马逊 CloudWatch,旨在增强您的卓越运营。首先,SageMaker 多模型终端节点使您能够在单个端点后面部署多种模型,从而减少需要管理的端点数量。SageMaker 会根据您的流量规律,管理模型的加载和缓存。您无需重新部署端点即可添加或更新模型。只需将模型上传到 SageMaker 托管的 Amazon S3 位置即可。此外,SageMaker 会自动与 CloudWatch 集成,您可以在 CloudWatch 中跟踪模型的指标、事件和日志文件,深入了解模型的性能。您还可以设置警报,在问题影响客户体验之前主动进行监控。

阅读“卓越运营”白皮书

API Gateway 提供内置机制对 API 请求进行身份验证和授权,防止拒绝服务攻击或其他可能使您的后端资源过载的滥用行为。您还可以使用亚马逊 Cognito 用户池、OAuth 2.0 或 IAM 角色来控制对您的 API 的访问权限。为了保护数据,API Gateway 会确保进入您端点的数据经过 SSL/TLS 加密。API Gateway 还支持 API 节流,有助于保护您的 API 免受过多流量或滥用的影响。 另外,可以考虑在 API 网关前面添加 Web 应用程序防火墙 AWS WAF,以保护应用程序免受基于 Web 的攻击和漏洞。最后,考虑使用 AWS Shield 来保护您的工作负载免受分布式拒绝服务 (DDoS) 攻击。

阅读《安全性》白皮书

本指南中部署了 API GatewayLambdaSageMaker,来提高工作负载的可靠性。首先,API Gateway 提供了内置的容错和自动扩缩功能,可应对流量激增。API Gateway 还与 LambdaSageMaker 集成,使您可以轻松构建可扩展的无服务器 API。此外,SageMaker 的目标是为运行机器学习工作负载和提供机器学习模型服务带来高可靠性和可用性。SageMaker 可实现托管的自动扩缩、容错、运行状况检查、监控和诊断,还可在分布于多个可用区的分布式基础设施上运行,来确保高可用性。这一切都确保了模型训练和推理的可靠性。

阅读《可靠性》白皮书

此处使用 SageMaker 来提高性能效率,实现可用于托管机器学习模型的高性能、低延迟推理服务。您可以轻松配置实例类型、数量和其他部署配置,以调整推理工作负载的规模,优化延迟、吞吐量和成本。

阅读《性能效率》白皮书

部署大量模型时,SageMaker 多模型终端节点是一种可扩展且经济实惠的方式。这些端点使用同一个容器来托管所有模型,减少了管理不同独立端点的开销。在某些模型使用率不高的情况下,您可以通过资源共享,以最大限度地提高基础设施的利用率,与使用单独的端点相比,还能节约成本。 

阅读《成本优化》白皮书

SageMaker 异步推理是一种对传入请求进行排队并异步处理这些请求的功能。这意味着,SageMaker 可以在不使用时自动缩减到零实例,从而在闲置时节省计算资源,并帮助最大限度地减少运行云工作负载对环境的影响。 

阅读《可持续性》白皮书

实施资源

示例代码为起点。它经过行业验证,是规范性但不是决定性的,可以帮助您开始。
GitHub 上的开放示例代码

相关内容

AWS 机器学习 | 博客

使用 Amazon 人工智能内容审核服务,打造安全的图像生成和扩散模型

这篇文章探讨了如何使用 AWS 人工智能服务 Amazon Rekognition 和 Amazon Comprehend 以及其他技术,近乎实时地有效审核 Stable Diffusion 模型生成的内容。

免责声明

示例代码;软件库;命令行工具;概念验证;模板;或其他相关技术(包括由我方人员提供的任何前述项)作为 AWS 内容按照《AWS 客户协议》或您与 AWS 之间的相关书面协议(以适用者为准)向您提供。您不应将这些 AWS 内容用在您的生产账户中,或用于生产或其他关键数据。您负责根据特定质量控制规程和标准测试、保护和优化 AWS 内容,例如示例代码,以使其适合生产级应用。部署 AWS 内容可能会因创建或使用 AWS 可收费资源(例如,运行 Amazon EC2 实例或使用 Amazon S3 存储)而产生 AWS 费用。

找到今天要查找的内容了吗?

请提供您的意见,以便我们改进网页内容的质量。