Tagged

TOOL

A collection of 155 posts

ComfyUI自定义节点开发
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 import Image