Tao Train Ocdnet

Wraps NVIDIA's TAO OCDNet model for arbitrary-oriented text detection in natural scenes using differentiable binarization. The skill handles train, evaluate, inference, export, prune, and quantize workflows with full AutoML support enabled by default. Data prep requires extracting archives into split folders with img/ and gt/ subdirectories, and every action needs explicit dataset path overrides since nothing unpacks at runtime. Training monitors hmean and outputs Lightning checkpoints you can route to pruning or TensorRT deployment. The AutoML integration is solid but requires disabling warmup epochs for single-epoch smoke runs to avoid trainer failures. If you need rotated bounding boxes for storefront signs or document photos, this does the job without making you build the pipeline from scratch.

Install to Claude Code

npx -y skills add nvidia/skills --skill tao-train-ocdnet --agent claude-code