> ## Content Index
> Fetch the complete content index at: https://www.hubwiz.com/blog/llms.txt
> Use this file to discover other available public pages before exploring further.

# ComfyUI-Autodistill图像自动标注
- URL: https://www.hubwiz.com/blog/comfyui-autodistill/
- Published: 2024-12-10T11:30:26.000Z
- Updated: 2024-12-11T01:55:33.000Z
- Description: ComfyUI-Autodistill是ComfyUI平台上的图像集自动标注节点包，利用基础模型的能力自动标注图像并生成目标检测数据集。
- Author: admin
- Tags: TOOL, COMFYUI, COMFYUI-AUTODISTILL, IMAGE ANNOTATION, #Import 2026-10-14 14:41

[ComfyUI-Autodistill](http://sc.hubwiz.com/codebag/comfyui-autodistill/?ref=hubwiz.com)是ComfyUI平台上的图像集自动标注节点包，利用基础模型的能力自动标注图像并生成目标检测数据集。

## 1、代码组织

ComfyUI-Autodistill节点包的目录组织说明如下：

| 文件路径                          | 文件说明                                |
| ----------------------------- | ----------------------------------- |
| requirements.txt              | 依赖包清单文件                             |
| \_\_init\_\_.py               | 包入口文件                               |
| py/                           | 自定义节点目录                             |
| py/caption\_ontology.py       | CaptionOntology节点，定义基础模型的提示和对应的标注信息 |
| py/load\_base\_model.py       | LoadBaseModel节点，加载基础模型              |
| py/empty\_dict.py             | EmptyDict节点，输出空字典对象                 |
| py/dict\_set\_value.py        | DictSetValue节点，设置字典对象的值             |
| py/label\_images.py           | LabelImages节点，自动标注指定的图像集            |
| py/visualize\_dataset.py      | VisualizeDataset节点，可视化生成的数据集        |
| examples/                     | 示例工作流目录                             |
| examples/auto-annotation.json | 自动标注示例工作流                           |
| images/                       | 待标注的示例图像集目录                         |
| dataset/                      | 完成标注的数据集目录                          |
| requirements.txt              | 依赖包清单文件                             |

## 2、安装

将节点包解压到 `ComfyUI/custom_nodes` 目录下，结果如下所示：

```
ComfyUI
  |- custom_nodes
        |- ComfyUI-Autodistill
              |- __init__.py
              |- ...

```

然后进入 `ComfyUI-Autodistill` 目录，执行如下命令安装依赖包：

|

```
pip install -r requirements.txt

```

## 3、使用示例数据集快速上手

在ComfyUI界面中，点击【Load】按钮载入 `examples/auto-annotation.json`工作流，如下图所示：

![automated annotation example workflow](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/auto-annotation.png)

上面工作流执行后，将自动在指定的输出文件夹得到标注好的数据集。点击预览图，可以看到瓶子和瓶盖 已经被标注了：

![mask annotation](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/mask-annotation.png)

`LabelImages`节点负责对指定的图像集进行自动标注并形成目标检测数据集，它需要一个基础模型作为输入， 同时也需要指定要标注的图像文件所在的目录。

`LoadBaseModel`节点负责加载指定类型的基础模型，同时也需要指定一个标注提示（CaptionOntology）以便 指导基础模型进行标注。

`CaptionOntology`节点负责定义提供给基础模型的标注提示，标注提示将提示映射为标签，例如这个节点表示 基础模型应当标注出图像中的 `milk bottle`，并在生成的数据集中设置标签为 `bottle`：

![ontology example](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/ontology-example.png)

`EmptyDict`节点创建一个空字典对象，`DictSetValue`节点设置字典对象的键/值对，因此下面的工作流片段：

![dict part](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/dict-part.png)

实际上创建了一个如下的字典对象：

```
{
  "milk bottle": "bottle",
  "blue cap": "cap",
}

```

`VisualizeDataset`节点负责显示生成的数据集中的前N个带标注信息的图像。

## 4、自定义节点手册

### 4.1 LabelImages

![label images node](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/label-images.png)

`LabelImages`节点利用基础模型对指定的图像文件进行自动标注，生成目标检测数据集。

输入：

- base\_model：基础模型，由 `LoadBaseModel`节点加载
- input\_folder：待标注的图像文件目录
- extension：待标注图像文件的后缀，默认：'png'
- output\_folder：待生成数据集的保存目录，包含原始图像和标注信息

输出：

- dataset：生成的目标检测数据集

### 4.2 LoadBaseModel

![load base model node](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/load-base-model.png)

`LoadBaseModel`节点按指定的ontology信息加载指定的基础模型。

输入：

- ontology：基础模型的提示与标注映射描述，由 `CaptionOntology`节点创建
- model：要加载的基础模型，默认："Grounded SAM"

输出：

- base\_model：已加载的基础模型

### 4.3 VisualizeDataset

![visualize dataset node](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/visualize-dataset.png)

`VisualizeDataset` 节点可视化生成的数据集中的前N个带标注图像。

输入：

- dataset：自动标注生成的数据集
- first\_n：要叠加标注信息的前多少个图像
- mask：是否叠加遮罩标注
- box：是否叠加包围盒标注
- label：是否叠加分类标签

输出：

- images：生成的带标注图像批

### 4.4 CaptionOntology

![caption ontology node](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/caption-ontology.png)

`CaptionOntology`节点定义用于指导基础模型的标注行为的提示信息。

输入：

- dict：提示/分类标签字典

输出：

- ontology：提示标注对象

### 4.5 EmptyDict

![empty dict node](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/empty-dict.png)

`EmptyDict`节点定义一个空字典。

输出：

- dict：空字典

### 4.6 DictSetValue

![dict set value node](http://sc.hubwiz.com/codebag/comfyui-autodistill/img/dict-set-value.png)

`DictSetValue`节点设置字典对象的键/值对。

输入：

- key：键
- value：值

---

原文链接：[ComfyUI-Autodistill节点包](http://sc.hubwiz.com/codebag/comfyui-autodistill/?ref=hubwiz.com)

汇智网编辑整理，转载请标明出处