项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 2798
使用教程
环境要求
- 已安装 Python 与 pip 的本地环境(README 未限定具体版本)
- 需同时安装 autodistill 主包和对应的 Base/Target 模型插件包
- 源码安装方式需本机可访问 GitHub 并执行 git clone
安装与启动步骤
-
1安装核心包
安装 autodistill 主包及 GroundedSAM、YOLOv8 插件。插件需与主包分开安装,否则无法使用对应模型。
pip install autodistill autodistill-grounded-sam autodistill-yolov8 -
2源码安装方式
可选。克隆仓库并以可编辑模式安装,适合本地开发调试,需先进入仓库目录。
git clone https://github.com/roboflow/autodistill cd autodistill pip install -e . -
3一行命令蒸馏
对 images 目录内图片用 Grounding DINO 自动标注,再用标注数据训练 YOLOv8,结果存到 dataset。
autodistill images --base="grounding_dino" --target="yolov8" --ontology '{"prompt": "label"}' --output="./dataset" -
4运行模型推理
README 示例中对训练好的目标模型推理。target_model 需先在代码中按插件说明初始化。
pred = target_model.predict("./dataset/valid/your-image.jpg", confidence=0.5) print(pred) -
5部署到 Roboflow
可选。将训练好的 YOLOv8 模型上传到 Roboflow 做边缘部署,需替换 API_KEY 与 PROJECT_ID。
from roboflow import Roboflow rf = Roboflow(api_key="API_KEY") project = rf.workspace().project("PROJECT_ID") project.version(DATASET_VERSION).deploy(model_type="yolov8", model_path=f"./runs/detect/train/")
关键配置
| 配置项 | 必填 | 说明 | 示例 |
|---|---|---|---|
--base | 是 | 指定用于自动标注的基础模型 | grounding_dino |
--target | 是 | 指定用自动标注数据训练的轻量目标模型 | yolov8 |
--ontology | 是 | 提示词与标签的映射,可指定多组 | {"prompt": "label"} |
--output | 是 | 标注结果与数据集的输出目录 | ./dataset |
api_key | 否 | Roboflow 部署时使用的 API 密钥 | API_KEY |
model_path | 否 | 部署时训练产物所在路径 | ./runs/detect/train/ |
如何确认成功
命令执行后在 ./dataset 目录生成已标注数据集并完成 YOLOv8 训练;或在代码中调用 predict 正常返回预测结果。
常见问题
Q:只安装 autodistill 主包够用吗?
A:不够。Autodistill 采用插件化设计,还需安装对应的 Base Model 和 Target Model 插件,如 autodistill-grounded-sam、autodistill-yolov8。
Q:不想在本地跑可以吗?
A:可以。README 提到可使用 Roboflow 托管的 Autodistill 版本在云端完成图像标注。
Q:目前支持哪些视觉任务?
A:README 说明当前支持目标检测和实例分割,未来计划扩展到语言等其他模型。
Q:哪些目标模型可以部署到 Roboflow?
A:README 列出 YOLOv8 目标检测/实例分割、YOLOv5 目标检测/实例分割均支持,YOLOv8 分类暂不支持。
Q:一行命令里的 prompt 和 label 怎么填?
A:prompt 是给基础模型的提示词,label 是保存的标签名,可成对指定多组,按需替换示例值。
注意事项
- 基础模型与目标模型以插件形式单独安装,可减少依赖与许可证冲突。
- 部署代码中的 API_KEY、PROJECT_ID、DATASET_VERSION 均为占位符,需替换为真实值。
- 完整教程可参考 README 中的 Colab Notebook 链接与官方文档站。
- 更多可用 Base/Target 模型见 README 的 Available Models 章节。
核心亮点
- 支持多种基础模型(如 DINO、CLIP)自动标注,覆盖检测、分割、分类任务
- API 简洁,几行代码完成标注到训练,上手快
- 与主流训练框架(如 Ultralytics)集成,方便部署
不足之处
- 自动标注质量依赖基础模型,复杂场景可能需人工修正
- 文档/社区待观察
适用场景
- 快速构建定制目标检测模型,如工业质检
- 缺乏标注数据的冷启动项目
- 自动化数据标注流水线集成
替代项目
Label Studio、Roboflow、H2O Label Gen
上一篇:ColossalAI
下一篇:CoOp
同类项目推荐
EAGLE
开源
让大模型生成速度翻倍,无需牺牲输出质量
Official Implementation of EAGLE-1 (ICML'24), EAGLE-2 (EMNLP'24), and EAGLE-3 (NeurI···
nnscaler
开源
写单卡模型,编译器自动生成多卡并行训练方案
nnScaler: Compiling DNN models for Parallel Training
Graph-CoT
开源
让大模型在知识图谱上一步步推理,提升复杂问答准确率
Graph Chain-of-Thought: Augmenting Large Language Models by Reasoning on Graphs (ACL···
jax-llm-examples
开源
纯JAX极简代码,跑通LLM训练与推理
Minimal yet performant LLM examples in pure JAX
KIVI
开源
2bit 量化 KV Cache,长上下文推理显存减半不降质
[ICML 2024] KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cache
hls-foundation-os
开源
微调 HLS 遥感基础模型,快速落地分类任务
This repository contains examples of fine-tuning Harmonized Landsat and Sentinel-2 (···
ml-engineering
开源
大模型训练推理踩坑指南,照着调优省半年时间
Machine Learning Engineering Open Book
terratorch
开源
一键微调遥感大模型,快速落地地理空间AI应用
A Python toolkit for fine-tuning Geospatial Foundation Models (GFMs).
SuperPiano
开源
在 Colab 里用 Transformer 一键生成钢琴曲
Absolutely amazing SOTA Google Colab (Jupyter) Notebooks for creating/training SOTA ···
