tao-train-deformable-detr - 训练和运行 TAO Deformable DETR
训练、评估、导出、量化 TAO Deformable DETR 二维目标检测模型并运行推理。
标签
更新于: 2026-09-28能力
典型输入
典型输出
该技能可以做什么
- 训练 Deformable DETR 模型
- 评估目标检测模型
- 运行模型推理
- 将模型导出为 ONNX
- 量化模型权重
- 生成 TensorRT 引擎
- 运行 AutoML 训练
输入
- COCO 或 COCO-raw 数据集
- 图像压缩包
- JSON 标注文件
- 类别映射文件
- 模型检查点
- 动作规范
- 规范覆盖参数
- 数据集类别元数据
输出
- 训练检查点
- 评估指标
- 推理预测结果
- 导出的 ONNX 模型
- 量化模型产物
- TensorRT 引擎文件
- 动作状态和日志
要求
- Docker
- NVIDIA Container Toolkit
- 用于生成 TensorRT 引擎的 TAO Deploy 容器
- 用于运行 AutoML 的已打包训练模式和模板