Agent skill · research science · nvidia

tao-train-ocrnet

OCRNet for scene text recognition. Recognizes text content from cropped text-region images and supports CTC

Why this skill is useful

Provides specific commands and configurations for training and deploying OCRNet models that the AI wouldn't reliably generate on its own.

What it needs

Requires docker, nvidia-container-toolkit installed locally. About 9k tokens when loaded. Last updated 2026-08-07. 2,816 stars on the source repository.

What this skill does

OCRNet Standalone install? If this session was not initialized by the TAO skill bank plugin, run the tao-setup skill first (host preflight, credentials, cross-skill discovery). OCRNet for scene text recognition. Recognizes text content from cropped text region images. Supports CTC and attention-based decoders. Set train.pretrainedmodelpath for pretrained OCR weights. For TAO Deploy TensorRT actions (gentrtengine, TensorRT evaluate, and TensorRT inference), read references/tao-deploy-ocrnet.md first. Deploy spec templates live in this skill's references/ folder with the spectemplatedeploy.yaml prefix. Dataclass Schemas Generated TAO Core schemas are packaged in schemas/<action>.schema.json, with schemas/manifest.json listing available actions. Each generated schema also emits references/spectemplate<action>.yaml from the schema top-level default field. AutoML enablement is declared at the model layer in references/skillinfo.yaml via automlenabled. Runnable AutoML for an action requires schemas/<action>.schema.json and references/spectemplate<action>.yaml to exist and parse. Use the packaged selected-action schema for automldefaultparameters, automldisabledparameters, defaults, min/max bounds, enums, option weights, math conditions, dependencies, and popular parameters. Do not expect ~/tao-core at runtime; maintainers regenerate schemas/templates before packaging the skill bank. Train Action Policy This model is AutoML-enabled at the model layer. Before handling any train-stage request, read references/skillinfo.yaml and resolve the run override from either an explicit automlpolicy value or the user's workflow request. Use automlpolicy: on by default and only expose on / off in new launch prompts. Treat phrases like "turn off AutoML", "disable AutoML", "no HPO", or "plain training" as automlpolicy: off for this run only. …

How to use it

Reference it in AdaL, Claude Code, Cursor or any coding agent — nothing to install:

@skills nvidia/tao-train-ocrnet

View the source on GitHub

Browse the @skills marketplace