tao-train-codetr - 训练和运行 TAO Co-DETR 目标检测
训练、评估并运行 TAO Co-DETR 目标检测模型的推理。
标签
更新于: 2026-10-05能力
典型输入
典型输出
该技能可以做什么
- 训练 Co-DETR 模型
- 评估检测模型
- 运行图像推理
- 加载检测检查点
- 应用类别映射
输入
- Hydra 实验规格文件
- COCO 图像数据集
- COCO 标注文件
- 推理图像目录
- 类别映射文件
- 检测器检查点
- 类别映射配置
输出
- 训练检查点
- 评估结果
- 推理检测结果
- 结果目录
要求
- Docker
- nvidia-container-toolkit
- TAO PyTorch 镜像
- 镜像中的 Co-DETR 软件包