tao-train-oneformer - 训练与部署 OneFormer 分割模型
训练、评估、导出、量化并推理支持全景、实例和语义分割的 TAO OneFormer 模型。
标签
更新于: 2026-10-06能力
典型输入
典型输出
该技能可以做什么
- 训练 OneFormer 模型
- 评估分割模型
- 将模型导出为 ONNX
- 量化模型权重
- 运行分割推理
- 配置 AutoML 训练
输入
- COCO 分割数据集
- 模型检查点
- 预训练骨干网络
- 动作规格
- 输出路径
输出
- 训练后的模型检查点
- 评估指标
- 导出的 ONNX 模型
- 量化后的模型
- 分割推理结果
要求
- Docker
- NVIDIA Container Toolkit
- TAO 运行环境