tao-train-codetr - Train and run TAO Co-DETR object detection
Trains, evaluates, and runs inference for TAO Co-DETR object detection models.
Tags
Updated: 2026-10-05Capabilities
Typical Inputs
Typical Outputs
What this skill does
- Train Co-DETR models
- Evaluate detection models
- Run image inference
- Load detector checkpoints
- Apply category mappings
Inputs
- Hydra experiment specification
- COCO image datasets
- COCO annotation files
- Inference image directory
- Classmap file
- Detector checkpoint
- Category mapping
Outputs
- Training checkpoints
- Evaluation results
- Inference detection results
- Results directories
Requirements
- Docker
- nvidia-container-toolkit
- TAO PyTorch image
- Co-DETR package in the image