支持的模型
我们的目标是让 autodistill 支持使用所有基础模型作为基础模型,并将大多数SOTA监督模型作为目标模型。我们首先专注于目标检测和分割任务,但计划很快推出分类支持!未来,我们希望 autodistill 也能用于超越计算机视觉的模型。
- ✅ - 完成(点击行/列标题以进入仓库)
- 🚧 - 进行中的工作
对象检测¶
| 基准 / 目标 | YOLOv8 | YOLO-NAS | YOLOv5 | DETR | YOLOv6 | YOLOv7 | MT-YOLOv6 |
|---|---|---|---|---|---|---|---|
| DETIC | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| GroundedSAM | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| GroundingDINO | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| OWL-ViT | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| SAM-CLIP | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| LLaVA-1.5 | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| Kosmos-2 | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| OWLv2 | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| Roboflow Universe Models (50k+ 预训练模型) | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| CoDet | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| VLPart | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| Azure 自定义视觉 | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| AWS Rekognition | ✅ | ✅ | ✅ | ✅ | 🚧 | ||
| 谷歌视觉 | ✅ | ✅ | ✅ | ✅ | 🚧 |
实例分割¶
| 基础 / 目标 | YOLOv8 | YOLO-NAS | YOLOv5 | YOLOv7 | Segformer |
|---|---|---|---|---|---|
| GroundedSAM | ✅ | 🚧 | 🚧 | ||
| SAM-CLIP | ✅ | 🚧 | 🚧 | ||
| SegGPT | ✅ | 🚧 | 🚧 | ||
| 快速SAM | 🚧 | 🚧 | 🚧 |
分类¶
| 基础 / 目标 | ViT | YOLOv8 | YOLOv5 |
|---|---|---|---|
| CLIP | ✅ | ✅ | 🚧 |
| MetaCLIP | ✅ | ✅ | 🚧 |
| DINOv2 | ✅ | ✅ | 🚧 |
| BLIP | ✅ | ✅ | 🚧 |
| ALBEF | ✅ | ✅ | 🚧 |
| FastViT | ✅ | ✅ | 🚧 |
| AltCLIP | ✅ | ✅ | 🚧 |
| 冬天 | 🚧 | 🚧 | 🚧 |
| 打开火鹤 | 🚧 | 🚧 | 🚧 |
| GPT-4 | |||
| PaLM-2 |
Roboflow模型部署支持¶
您可以选择在Roboflow上部署一些使用Autodistill训练的目标模型。在Roboflow上部署可以让您使用一系列简洁的SDK,在边缘使用您的模型,从roboflow.js用于Web部署到NVIDIA Jetson设备。
以下Autodistill目标模型由Roboflow支持用于部署:
| 模型名称 | 支持吗? |
|---|---|
| YOLOv8目标检测 | ✅ |
| YOLOv8 实例分割 | ✅ |
| YOLOv5目标检测 | ✅ |
| YOLOv5 实例分割 | ✅ |
| YOLOv8 分类 |