TOOL
ComfyUI自定义节点开发
ComfyUI 是 Stable Diffusion 的强大而灵活的用户界面，允许用户通过基于节点的系统创建复杂的图像生成工作流程。虽然 ComfyUI 带有各种内置节点，但其真正的优势在于可扩展性。自定义节点使用户能够添加新功能、集成外部服务并根据其特定需求进行定制。 在这篇博文中，我们将介绍使用 ComfyUI 创建用于图像字幕的自定义节点的过程。此节点将以图像作为输入，并使用外部 API 返回生成的字幕。 我们将使用 Google Gemini API 生成图像的字幕。 1、自定义节点的完整代码 这是使用 Gemini API 执行 ImageCaptioning 的完整代码。 你可以将以下代码复制到 ComfyUI 中 custom_nodes 文件夹下的任何文件中，我将我的文件命名为 gemini-caption.py ： 生成图像标题的完整代码： import numpy as np from PIL