AutoVision¶
企业级视觉模型训练框架:自监督预训练 / 蒸馏 → 微调 → 评测 → 导出一条链走完。
Python API 与 autovision-cli(31 条命令)共享同一套 Pydantic 严格校验,覆盖
检测、旋转框、实例 / 全景 / 语义分割、姿态、深度、分类的完整任务面。
框架的三条差异化主线:
- 许可干净:Apache-2.0;全仓溯源门禁与权重许可门控(受限许可权重默认 不下载,AGPL 组件仅 extras 隔离),模型库页面对每个权重给出许可与 sha256。
- 可验证一致性:ONNX / TensorRT / OpenVINO 导出默认执行数值对拍门禁, 并输出可移植的后处理规格;官方权重加载与前向对拍(含 bit-equal); 金标准 loss 轨迹快照。
- 协议化基准:基准表只引官方发表值、自家数字必附复现 脚本、只增不改 + 勘误节。
快速开始¶
不训练,直接加载发布权重并预测:
import autovision
model = autovision.detection("rfdetr/rf-detr-small", pretrained=True)
result = model.predict("image.jpg") # {'bboxes': ..., 'scores': ..., 'labels': ...}
model.export("model.onnx") # parity 门禁 + postprocess 规格
可用模型与导出边界:autovision-cli list-models --task detection --pretrained、
autovision-cli model-info rfdetr/rf-detr-small。
目标检测,训练 → 验证 → 预测 → 导出:
import autovision
data = autovision.ensure_coco8() # 合成冒烟数据集;换 COCO/YOLO 配置、YAML 路径或字面量 "coco8"
model = autovision.detection("rtdetr/resnet18").train(
out="runs/det1", data=data, steps=2000, batch_size="auto",
)
model.val(data=data) # {'map': ..., 'map_50': ...}
model.predict("image.jpg") # {'bboxes': ..., 'scores': ..., 'labels': ...}
model.export("model.onnx") # 按 .onnx / .engine / .pt 后缀分发
等价命令行(同一套 Pydantic 校验):
autovision-cli train-object-detection out=runs/det1 data=data.yaml steps=2000
autovision-cli val model=runs/det1/exported_models/exported_best.pt data=coco8
autovision-cli export model=runs/det1/exported_models/exported_best.pt out=model.onnx precision=fp16
autovision-cli predict model=runs/det1/exported_models/exported_best.pt source=clip.mp4 out=preds/ batch_size=4
安装¶
Python ≥ 3.10。源码安装是当前可用路径;PyPI 发布后可用
pip install autovision-framework(模型包按 extras 惰性加载,缺失时
对应清单为空而不是报错):
pip install -e ".[transformers,timm,export]" # 常用: HF 骨干 + timm + ONNX 导出
pip install -e ".[rfdetr,export]" # 预训练 RF-DETR 直接推理
pip install -e ".[video]" # 参考推理运行时(视频/RTSP, OpenCV)
pip install -e ".[dev,export,serving,video,transformers]" # 开发/全量测试
PyPI 发行名计划使用 autovision-framework,发布后可用。
GTR 全族(MIT)需要 gtr extra;完整 extras 表见仓库 README。预训练权重从
HuggingFace 下载,受限网络可设镜像;受限许可权重默认不下载(DINOv2
CC-BY-NC 4.0 / DINOv3 License 需显式自证接受条款),许可边界见仓库 NOTICE。
各类权重的下载命令、缓存与镜像配置见权重下载。
能力速览¶
| 子系统 | 能力 |
|---|---|
| 配置 / 注册 | Pydantic 严格校验 + did-you-mean 报错 + 注册表;CLI 与 API 同一套校验 |
| 模型层 | 7 个注册源(torchvision / timm / super_gradients / ultralytics(AGPL 隔离)/ dinov2 / dinov3 / rfdetr)+ 树内 rtdetr/*、detr/*、dfine/*、picodet/*、rfdetr/*、gtr/* |
| 数据 / 变换 | 11 类数据集;mmap 文件清单(内容指纹 / 原子发布);检测增广全套(mosaic / mixup / copyblend / box-aware eraser);VOC / YOLO-pose / YOLO-seg 转换 + COCO 合并 |
| 预训练方法 | DistillationV3、SimCLR、DINO、DINOv2、DINOv3、DenseCL、DetCon B/S、训练中 kNN 探针 |
| 运行时 | DDP、混合精度、原子 checkpoint + resume、NaN 捕获、JSONL / TB / MLflow / W&B、OTel(opt-in) |
| 任务层 | 8 个任务面;任务指标支持跨 rank 汇聚(DDP 验证);误差分析(FP 归因 + 混淆矩阵 + top-FP/FN 画廊)+ 伪标注 / 自训练闭环;超参搜索(随机 / 网格 / ASHA + 报告);任务感知可视化(框 / mask / 旋转框 / 姿态 / 语义与全景分割 / 深度) |
| 部署 | ONNX(FP16 黑名单 + 数值对拍门禁)、TensorRT、int8 QDQ、benchmark 三后端 parity、参考推理运行时(视频 / RTSP + 三段式管线 + batch,同源前后处理;可选 ByteTrack 跟踪、LineZone / PolygonZone 计数、轨迹平滑;mask 默认 RLE)、参考 serving(FastAPI + TLS/多 worker + 限流 + 漂移 + A/B + /metrics) |
各模型的许可 / sha256 / 官方指标 / 一行加载代码见模型库。
文档导航¶
| 页面 | 内容 |
|---|---|
| 模型库 | 每模型:许可 / sha256 / 官方指标 / 加载代码(机器生成) |
| 基准 | 精度与延迟基准(不可变协议 + 勘误节) |
| 导出与部署 | 动态 H/W / int8 / TensorRT 11 / parity / GTR 部署注记 |
| 插件协议 | entry-point 插件、API 版本、model_source 接入与模板 |
| 私有注册表 | current / rollback / audit / 部署门禁 |
| 合规报告 | 依赖许可 / 模型卡 / 数据快照证据包 |
| 硬件伙伴包 | 一伙伴一包、转换与目标端 parity 契约 |
| LTS 政策 | LTS 节奏、支持窗口、安全 backport |
| 用户自持权重 | 仓库外 DINOv3 检测器的加载路线与许可边界 |
| 已知限制 | 定位、适配缺口、谱系状态、导出边界 |
| 架构总览 | as-built 分层契约 / 模块地图 / 质量门禁 |
| 贡献指南 | 开发环境 / 提交前门禁 / 许可与溯源红线 |
| 发布流程 | 版本政策 / 发布清单 / 权重发布政策 |
| 模型发布 | 许可审计 / 模型卡 / HF 上传 / 注册表 / 发布门禁 |
许可¶
Apache-2.0(全文:https://www.apache.org/licenses/LICENSE-2.0;仓库根
LICENSE)。第三方组件与权重溯源见仓库根 NOTICE;交互式命令全表:
autovision-cli --help
autovision-cli list-models # 当前环境可用的模型清单(按已安装 extras 过滤)