使用 Imagen on Vertex AI 构建 AI 图片生成器应用
实验
15 分钟
universal_currency_alt
免费
show_chart
入门级
info
此实验可能会提供 AI 工具来支持您学习。
bb-ide-genai-002

概览
- 实验是计时的,并且无法暂停。计时器在您点击开始实验后即开始计时。
- 随附的云终端已预先配置了 gcloud SDK。
- 使用终端执行命令,然后点击检查我的进度以验证您的实验成果。
目标
借助 Vertex AI 上的生成式 AI,您可以访问 Google 的大型生成式 AI 模型,对模型进行测试、调整并部署到 AI 赋能的应用中。在本实验中,您将执行以下操作:
-
连接到 Vertex AI(Google Cloud AI 平台):了解如何使用 Vertex AI SDK 建立与 Google AI 服务的连接。
-
加载预训练的图片生成模型:了解如何使用强大的预训练 AI 模型,而无需从头构建。
-
向 AI 模型发送文本:了解如何为 AI 提供输入。
-
提取基于图片的 AI 回答:了解如何处理和解读 AI 模型生成的图片回答。
-
了解构建 AI 应用的基础知识:深入了解将 AI 集成到软件项目中的核心概念。
使用生成式 AI
开始实验后,您将看到一个分屏视图,左侧是代码编辑器,右侧是实验说明。按照说明步骤使用 Vertex AI Python SDK 与生成式 AI API 进行交互。
- 点击文件 -> 新建文件,在代码编辑器中打开一个新文件。
- 复制提供的代码段并将其粘贴到文件中。
import argparse
import vertexai
from vertexai.preview.vision_models import ImageGenerationModel
def generate_image(
project_id: str, location: str, output_file: str, prompt: str
) -> vertexai.preview.vision_models.ImageGenerationResponse:
"""Generate an image using a text prompt.
Args:
project_id: Google Cloud project ID, used to initialize Vertex AI.
location: Google Cloud region, used to initialize Vertex AI.
output_file: Local path to the output image file.
prompt: The text prompt describing what you want to see."""
vertexai.init(project=project_id, location=location)
model = ImageGenerationModel.from_pretrained("imagen-3.0-generate-002")
images = model.generate_images(
prompt=prompt,
# Optional parameters
number_of_images=1,
seed=1,
add_watermark=False,
)
images[0].save(location=output_file)
return images
generate_image(
project_id='{{{ project_0.project_id | "project-id" }}}',
location='{{{ project_0.default_region | "REGION" }}}',
output_file='image.jpeg',
prompt='Create an image of a cricket ground in the heart of Los Angeles',
)
- 依次点击文件 > 保存,在“名称”字段中输入
GenerateImage.py
,然后点击保存。
- 点击代码编辑器中的三角形图标,或在代码编辑器窗格内的终端中调用以下命令,以执行 Python 文件。这将生成一个名为
image.jpeg
的图片文件。
/usr/bin/python3 /GenerateImage.py
- 现在,点击探索器 > image.jpeg,查看生成的图片
代码说明
- 此代码段会加载 Vertex AI 的 ImageGenerationModel (imagen-3.0-generate-002) 预训练 AI 模型。
- 该代码会调用已加载的 Gemini 模型的
generate_image
方法。
- 该方法的输入是一个文本提示。
- 该代码使用 Gemini 理解文本提示并构建 AI 图片。
注意:默认情况下,系统会在图片中添加 SynthID 水印,但您可以通过指定可选参数 add_watermark=False
来停用此功能。您不能同时使用种子值和水印。详细了解 SynthID 水印
亲自尝试一下吧! 尝试使用不同的提示,探索 Gemini 的功能。
点击检查我的进度以验证是否完成了以下目标:
向生成式 AI 发送文本提示请求并接收回答
恭喜!
您已完成本实验!恭喜!
版权所有 2025 Google LLC 保留所有权利。Google 和 Google 徽标是 Google LLC 的商标。其他所有公司名和产品名可能是其各自相关公司的商标。