ComfyUI-DataSet 插件新手教程

1. 插件简介

ComfyUI-DataSet 是一个专门为AI绘画师和模型训练者设计的数据处理工具箱。插件地址：https://github.com/daxcay/ComfyUI-DataSet

这个插件就像一个专业的图片和文字管理小助手，它能帮你整理、分析和处理用于训练AI模型的数据。简单来说，它可以：

• 帮你批量处理图片的描述文字（就像给每张图片贴标签）
• 分析这些标签词的使用频率（哪些词用得最多）
• 自动生成各种统计图表（词云图、网络关系图等）
• 批量复制、移动、整理文件
• 利用AI聊天机器人生成图片描述

2. 如何安装

方法一：使用 Comfy Manager（推荐新手）

1. 打开 ComfyUI，点击界面右侧的"Manager"按钮
2. 点击"Custom Nodes Manager"
3. 在搜索框里输入"DataSet"
4. 找到这个插件并点击安装
5. 重启 ComfyUI 就可以使用了

方法二：手动安装

1. 找到你的 ComfyUI 安装目录下的 Custom Nodes 文件夹
2. 在这个文件夹里打开命令行（按住Shift键右键点击空白处，选择"在此处打开命令行"）
3. 输入：git clone https://github.com/daxcay/ComfyUI-DataSet.git
4. 等待下载完成后，进入 ComfyUI-DataSet 文件夹
5. 输入：pip install -r requirements.txt
6. 重启 ComfyUI

3. 节点详解

3.1 DataSet_Visualizer 节点 - 数据可视化小助手

这个节点就像一个图表制作工具，能把你的文字标签变成好看的图表。

参数详解：

输出说明：

• 图表路径：生成的图片文件保存在哪里
• 图表图片：可以直接在ComfyUI里预览的图片

3.2 DataSet_CopyFiles 节点 - 文件复制小能手

这个节点就像一个智能文件管理员，能帮你批量复制文件。

参数详解：

3.3 DataSet_TriggerWords 节点 - 触发词提取器

这个节点就像一个聪明的筛子，专门找出包含字母和数字的特殊词汇。

参数详解：

输出说明：

• 提取的词汇：找到的所有触发词

3.4 DataSet_TextFilesLoad 节点 - 文本文件加载器

这个节点就像一个文件阅读器，能一次性读取多个文本文件的信息。

参数详解：

输出说明：

• 文件名：所有文件的完整名称
• 文件名（无扩展名）：不带.txt的文件名
• 文件路径：每个文件的完整路径
• 文件内容：每个文件里的文字内容

3.5 DataSet_TextFilesLoadByDirectory 节点 - 文件夹文本加载器

这个节点就像上面那个的升级版，直接指定文件夹就能读取里面所有的文本文件。

参数详解：

输出说明：

• 文件名：文件夹里所有文本文件的名称
• 文件名（无扩展名）：不带.txt的文件名
• 文件路径：每个文件的完整路径
• 文件内容：每个文件里的文字内容

3.6 DataSet_TextFilesSave 节点 - 文本文件保存器

这个节点就像一个文档管理员，能把文字内容保存成文件。

参数详解：

3.7 DataSet_FindAndReplace 节点 - 查找替换工具

这个节点就像Word里的查找替换功能，能批量修改文本内容。

参数详解：

输出说明：

• 修改后的文本：完成查找替换后的文字内容

3.8 DataSet_PathSelector 节点 - 路径选择器

这个节点就像一个配对专家，能找出哪些图片缺少对应的文字描述文件。

参数详解：

输出说明：

• 选中的文件名（带扩展名）：找到的配对文件完整名称
• 选中的文件名（无扩展名）：找到的配对文件名称（不带后缀）
• 选中的文件路径：找到的配对文件完整路径

3.9 DataSet_ConceptManager 节点 - 概念管理器

这个节点就像一个标签编辑器，能在文字描述里添加或删除特定的词汇。

参数详解：

输出说明：

• 修改后的文本：完成添加或删除操作后的文字内容

3.10 DataSet_OpenAIChat 节点 - 文字AI聊天助手

这个节点就像一个AI写手，能帮你生成各种文字内容。

参数详解：

输出说明：

• 生成的文本：AI生成的文字内容

3.11 DataSet_LoadImage 节点 - 图片加载器

这个节点就像一个图片管理员，能加载图片并提供图片的各种信息。

参数详解：

输出说明：

• 图片：加载的图片数据
• 遮罩：图片的遮罩信息
• 文件名：图片的完整文件名
• 文件名（无扩展名）：不带后缀的文件名
• 文件路径：图片的完整路径
• 目录路径：图片所在的文件夹路径

3.12 DataSet_SaveImage 节点 - 图片保存器

这个节点就像一个图片归档员，能批量保存图片。

参数详解：

3.13 DataSet_OpenAIChatImage 节点 - 图片AI描述生成器

这个节点就像一个AI图片解说员，能看图片并生成文字描述。

参数详解：

输出说明：

• 图片描述：AI生成的图片描述文字

3.14 DataSet_OpenAIChatImageBatch 节点 - 批量图片AI描述生成器

这个节点就像上面那个的升级版，能一次性处理多张图片。

参数详解：

输出说明：

• 图片描述列表：每张图片对应的AI生成描述

4. 使用技巧和建议

4.1 工作流程建议

1. 数据准备阶段：使用 DataSet_TextFilesLoadByDirectory 批量加载文本文件
2. 数据分析阶段：使用 DataSet_Visualizer 生成统计图表，了解数据分布
3. 数据清理阶段：使用 DataSet_FindAndReplace 批量修正错误
4. 数据增强阶段：使用 DataSet_ConceptManager 添加或删除标签

4.2 性能优化建议

• 处理大量文件时，建议分批处理，避免一次性加载过多数据
• 使用 AI 节点时，合理设置 token_length 参数，避免不必要的开销
• 定期清理生成的临时文件，保持文件夹整洁

4.3 API 使用建议

• 将 API 密钥设置为系统环境变量，提高安全性
• 环境变量名称：OPENAI_API_KEY, GROQ_API_KEY, ANTHROPIC_API_KEY
• 根据需要选择合适的 AI 模型，平衡效果和成本

5. 常见问题解答

Q1：为什么我的文件加载不成功？

A1： 检查文件路径是否正确，确保文件存在且格式正确。文本文件必须是 .txt 格式。

Q2：生成的图表在哪里？

A2： 图表会自动保存到 ComfyUI 的输出目录中，同时也会通过输出端口提供图片数据。

Q3：API 密钥应该怎么设置？

A3： 可以直接在节点中输入，也可以设置为系统环境变量。推荐使用环境变量方式，更安全。

Q4：为什么 AI 生成的描述不准确？

A4： 尝试调整提示词，提供更明确的指导。也可以选择更高级的模型，或者调整图片细节参数。

Q5：如何处理中文文本？

A5： 插件支持中文，但要确保文本文件使用 UTF-8 编码保存。

6. 进阶应用场景

6.1 模型训练数据准备

结合多个节点创建完整的数据预处理流程：

1. 使用 DataSet_PathSelector 找出缺少标注的图片
2. 使用 DataSet_OpenAIChatImageBatch 批量生成图片描述
3. 使用 DataSet_ConceptManager 添加特定的训练标签
4. 使用 DataSet_Visualizer 验证数据质量

6.2 数据集质量分析

通过可视化工具深入了解数据集特征：

1. 生成词云图查看最常用的标签
2. 分析网络关系图了解标签之间的关联
3. 使用频率图找出可能的标签不平衡问题

6.3 批量数据处理

高效处理大量文件：

1. 使用 DataSet_CopyFiles 整理文件结构
2. 使用 DataSet_FindAndReplace 批量修正标注错误
3. 使用 DataSet_TextFilesSave 保存处理结果

这个插件为ComfyUI用户提供了强大的数据处理能力，无论是新手还是专业用户，都能从中受益。通过合理组合这些节点，可以大大提高数据处理的效率和质量。