命令行界面

您可以使用Autodistill通过命令行接口（CLI）。

CLI 允许您对模型进行推断或自动给一组图像加标签，而无需编写标签脚本。

安装¶

要安装CLI，请安装Autodistill Python包：

pip install autodistill

CLI接受多个参数：

• images: 您想要标记的图像文件夹的路径。
• --base: 你想用于标记的基础模型。这可以是来自Autodistill Model Zoo的任何模型。
• --target: 您希望使用的目标模型，以便利用您的带标签数据集训练模型。
• --ontology：您想要用于标记的本体。这必须是将发送到模型的文本提示与您希望保存在数据集中的标签之间的映射。例如，{"acoustic guitar": "guitar"} 将文本提示 acoustic guitar 发送到模型，然后将标签保存为 guitar 在您的数据集。
• --output: 您要保存标记数据集的文件夹路径。

这是一个示例：

autodistill images --base="grounding_dino" --target="yolov8" --ontology '{"prompt": "label"}' --output="./dataset"

此命令将使用 Grounding DINO 给名为 images 的目录中的所有图像贴标签，并使用带标签的图像训练 YOLOv8 模型。Grounding DINO 会用“提示”标签所有图像，并将标签保存为“标签”。您可以指定任意数量的提示和标签。生成的数据集将保存在名为 dataset 的文件夹中。