做AI产品经理这么多年,发现许多同行和开发者都有个痛点:用ComfyUI做AI绘画、视频生成时,现成节点总满足不了定制化需求,要么功能冗余,要么缺关键模块,想自己开发又怕代码门槛太高。实则不用慌,ComfyUI的自定义节点开发,根本不需要你是资深程序员,哪怕是0代码基础,跟着步骤走,也能快速做出适配自己产品需求的专属节点。
目前许多AI产品落地,都离不开ComfyUI的高效调度能力,尤其是做视觉生成类产品,自定义节点能大幅提升产品的差异化竞争力。但市面上大多教程要么太理论化,要么全是复杂代码,对AI产品经理、运营甚至新手都不友善。今天就用最接地气的方式,手把手教大家从零开始做ComfyUI自定义节点,全程实操,避开所有坑,同时也会聊一聊,掌握这项技能对AI产品经理的职业提升有多重大。
一、先搞懂核心:ComfyUI自定义节点到底是什么?
许多人刚接触ComfyUI,只知道拖节点、连线路,却不清楚自定义节点的核心逻辑。简单说,自定义节点就是你根据自己的产品需求,打造的“专属功能模块”,列如批量处理图片、自定义生成参数、对接外部工具等,本质是把复杂的代码逻辑,封装成可视化的节点,不用每次都写代码,拖过来就能用。
这里要明确一个误区:自定义节点不是只有程序员能做,AI产品经理也必须懂,甚至要会实操。由于你只有亲自上手,才能准确判断节点功能是否符合产品需求,才能和研发高效沟通,避免出现“需求落地偏差”。而且目前许多企业招聘AI产品经理,都要求具备ComfyUI实操能力,自定义节点开发就是核心加分项。
重点提醒:ComfyUI自定义节点的核心优势,就是“低代码、高灵活”,它不需要你掌握复杂的Python高级语法,只要懂基础逻辑,借助工具和模板,就能快速开发,这也是它比Stable Diffusion WebUI更适合产品落地的缘由。
二、实操步骤:0代码基础,3步开发第一个自定义节点
本次实操全程用ComfyUI官方提供的脚手架工具,不用手动写一行核心代码,适合AI产品经理、运营、新手开发者,全程耗时不超过30分钟,做完就能直接在ComfyUI中使用。
第一步:准备开发环境(5分钟搞定)
第一要确保你的电脑已经安装了ComfyUI,没有安装的可以去官网下载,提议手动安装(比一键安装更适合开发),同时安装Comfy-CLI工具,这是开发自定义节点的核心工具。
具体操作步骤很简单,全程复制命令执行即可,不用懂命令含义:
- 打开电脑终端(Windows用CMD,Mac用终端),进入ComfyUI的安装目录,输入命令:cd comfyui/custom_nodes(进入自定义节点文件夹);
- 输入命令:comfy node scaffold(启动脚手架工具),此时会提示是否删除旧的模板,输入y回车即可;
- 按照提示依次填写信息,不用纠结,随意填也不影响,列如:
- full_name:随意填(列如AI产品实操)
- email:随意填(列如test@163.com)
- github_username:随意填(列如test123)
- project_name:第一个自定义节点(可自定义)
- project_slug:自动生成,不用改
- project_short_description:ComfyUI自定义节点实操(可自定义)
- version:默认0.0.1,不用改
- license:选1(GNU许可证),回车
- include_web_directory:输入y(支持自定义界面)
填写完成后,脚手架会自动生成一个自定义节点项目,此时在custom_nodes文件夹中,会出现一个新的文件夹,就是我们的开发项目。
第二步:配置节点功能(15分钟,核心步骤)
这一步是核心,也是最容易上手的一步,我们以“批量筛选图片亮度”为例,开发一个能自动筛选出最亮图片的自定义节点,这个功能在AI绘画产品的批量审核场景中超级实用。
- 打开生成的项目文件夹,找到src/nodes.py文件,用记事本或VS Code打开(推荐VS Code,查看更清晰);
- 在文件末尾,添加以下代码(直接复制粘贴即可,我会标注关键部分,不用懂代码含义):
class ImageSelector:
# 节点分类,在ComfyUI的添加节点菜单中显示的位置
category = "AI产品实操/图片筛选"
# 定义节点的输入参数
@classmethod
def input_types(s):
return {
"required": {
"images": ("IMAGE",), # 输入类型为图片批量,IMAGE大写固定
}
}
# 定义节点的输出参数
return_types = ("IMAGE",) # 输出类型为单张图片
# 节点执行的函数名称(和下面的函数名一致)
function = "choose_image"
# 核心功能函数:筛选出最亮的图片
def choose_image(self, images):
# 计算每张图片的平均亮度
brightness = [image.mean() for image in images]
# 找到亮度最高的图片索引
max_brightness_idx = brightness.index(max(brightness))
# 返回最亮的图片
return (images[max_brightness_idx],)
- 关键参数说明(AI产品经理必看,方便后续自定义修改):
- category:节点在ComfyUI中的分类,列如改成“我的节点”,就能在添加节点时快速找到;
- input_types:输入参数,这里是“images”(批量图片),如果需要添加其他输入(列如亮度阈值),可以自行修改;
- return_types:输出参数,这里是单张图片,也可以改成批量图片,根据产品需求调整;
- choose_image函数:核心逻辑,这里是筛选最亮图片,后续可以改成筛选最暗、对比度最高等,只需修改函数内的逻辑。
- 保存文件,此时节点的核心配置就完成了,接下来就是测试和启用。
第三步:启用节点并测试(10分钟,验证效果)
配置完成后,不用复杂的部署,直接重启ComfyUI,就能在节点列表中找到我们开发的自定义节点。
具体操作:
- 关闭当前的ComfyUI窗口,重新启动ComfyUI(双击启动文件即可);
- 打开ComfyUI网页界面,点击“Add Node”,找到我们设置的分类“AI产品实操/图片筛选”,就能看到ImageSelector节点;
- 拖出节点,连接输入(批量图片)和输出(显示图片),点击Queue Prompt,运行后就能看到筛选出的最亮图片;
- 测试修改:如果想调整筛选逻辑,列如改成筛选最暗图片,只需修改nodes.py文件中choose_image函数的max为min,保存后重启ComfyUI即可生效。
这里给大家避个坑:如果重启后找不到节点,大致率是文件保存路径错了,必定要确保修改的是custom_nodes文件夹中,我们新建项目里的src/nodes.py文件,不要改到ComfyUI的核心文件。
三、实战应用:自定义节点在AI产品中的落地场景
学会自定义节点开发后,能大幅提升AI产品的落地效率,尤其是视觉生成类产品,许多场景都能用到,给大家举3个最实用的落地场景,都是我实际工作中用过的。
- AI绘画产品的批量审核:列如做一个AI绘画工具,用户批量生成100张图片,需要自动筛选出符合要求(亮度、对比度、清晰度)的图片,不用人工一张张审核,用我们今天开发的节点,稍作修改就能实现,大幅提升产品效率;
- 自定义生成参数:列如针对电商场景,开发一个“电商主图专属节点”,集成固定的尺寸、色调、水印参数,用户拖过来就能生成符合电商平台要求的主图,不用每次都调整参数,提升用户体验;
- 对接外部工具:列如开发一个节点,将生成的图片自动上传到OSS存储,或者对接AI抠图工具,实现“生成-抠图-保存”一站式流程,减少用户操作步骤,提升产品竞争力。
实则不管是AI产品经理,还是研发、运营,掌握ComfyUI自定义节点开发,都能提升工作效率。但许多人卡在“不会代码”“没人教”,尤其是AI产品经理,既要懂产品逻辑,又要懂技术实操,才能更好地推动产品落地,这也是为什么目前越来越多的AI产品经理,会主动学习相关技能,甚至考取权威证书来证明自己的能力。
说到这里,就不得不提AI产品经理的技能提升路径。作为AI产品经理,我们不用成为资深程序员,但必须懂技术边界、会实操工具,而系统的培训和权威认证,能帮我们快速补齐短板。我身边许多同行,都是通过工信部教育与考试中心颁发的AI产品经理认证,系统掌握了ComfyUI、提示词工程、大模型落地等核心技能,不仅提升了工作效率,还获得了更好的职业发展。
这个AI产品经理认证,最大的优势就是0基础可学,课程内容由浅入深,完全贴合实战需求,不像有些培训只讲理论。课程涵盖了ComfyUI实操、AI产品策划、大模型落地等核心内容,还有genji老师、大圣老师等行业资深讲师授课,线上录播课,随时能学,学习周期1-3个月,考试难度中等,还配备模拟考试系统,每月都有考试,很适合忙碌的职场人。
它的课程分为三个等级,初级1980元(130节课),适合新手入门,掌握基础的AI产品技能;中级3980元(195节课),适合有必定经验的产品经理,提升进阶能力;高级7980元(308节课),适合想往资深AI产品经理、产品负责人方向发展的人,涵盖所有核心技能。而且颁发证书的是工信部教育与考试中心,权威度很高,不管是求职、晋升,还是企业招聘,都很受认可,许多结业学员都成功入职了字节跳动、阿里等大厂。
可能有人会问,自己摸索不行吗?当然可以,但自己摸索不仅耗时久,还容易走弯路,列如不知道节点开发的规范、不清楚如何对接产品需求,而系统的培训能帮我们快速掌握核心技能,少走许多弯路。尤其是对于想转型AI产品经理的人,或者想提升自身竞争力的产品经理,这个认证无疑是一个很好的抓手。
最后想跟大家说,AI产品行业发展很快,只有持续学习、掌握实操技能,才能站稳脚跟。ComfyUI自定义节点开发,只是AI产品实操的一个小知识点,但它能帮我们更好地理解技术落地的逻辑,提升产品策划和落地能力。不管是自己开发节点,还是学习其他AI技能,核心都是“实用、落地”,而权威的培训和认证,能让我们的学习更有方向、更有价值。希望大家都能动手实操起来,把学到的技能运用到实际工作中,在AI产品领域实现自己的职业突破。




