AIKraft/AI/Войти
AIKraft/Skills/TAO обучение OCRNet

TAO обучение OCRNet

OCRNet for scene text recognition. Recognizes text content from cropped text-region images and supports CTC and attention-based decoders. Use when training, evaluating, exporting, pruning, quantizing, retraining, or running inference for a TAO OCRNet model. Trigger phrases include

nvidia official Дизайн

Что делает навык

OCRNet предназначен для распознавания текста на сцене по обрезанным изображениям текстовых областей, поддерживая декодеры CTC и на основе внимания. Skill позволяет обучать, оценивать, экспортировать, прунить, квантовать, дообучать модель и выполнять инференс. Для развертывания с использованием TAO Deploy TensorRT необходимо предварительно ознакомиться с документацией в папке references.

Когда применять

  • Обучение модели OCRNet с использованием AutoML или прямого обучения.
  • Выполнение инференса и оценки качества распознавания текста.
  • Экспорт модели, а также её прунинг и квантование для оптимизации.
  • Развертывание модели с помощью TAO Deploy TensorRT (генерация движка, оценка и инференс).

Как работает

  1. Если сессия не инициализирована плагином TAO skill bank, выполнить навык tao-setup для проверки хоста, учетных данных и обнаружения навыков.
  2. Для действий обучения (train) прочитать references/skill_info.yaml и определить политику AutoML (по умолчанию automl_policy: on).
  3. При включенном AutoML и наличии файлов schemas/train.schema.json и references/spec_template_train.yaml направить действие через tao-skill-bank:tao-run-automl.
  4. При отключенном AutoML (фразы «turn off AutoML», «disable AutoML», «no HPO», «plain training») или отсутствии схем выполнить прямое обучение модели.
  5. Для развертывания TensorRT прочитать references/tao-deploy-ocrnet.md и использовать шаблоны спецификаций из папки references/ с префиксом spec_template_deploy_*.yaml.
  6. Использовать сгенерированные схемы TAO Core из schemas/<action>.schema.json для получения параметров, границ и зависимостей.

Примеры запросов агенту

Обучи OCRNet с включенным AutoML.
Выполни инференс для модели OCRNet.
Сгенерируй TensorRT движок для OCRNet.

Что понадобится

  • Инициализация сессии через навык tao-setup (если не выполнен плагином).
  • Наличие файлов schemas/<action>.schema.json и references/spec_template_<action>.yaml для работы AutoML.
  • Чтение references/tao-deploy-ocrnet.md перед действиями TAO Deploy TensorRT.

Описание составлено по SKILL.md навыка, сверено 05.10.2026.

Как подключить

1Скачать навык
# возьмите папку навыка «tao-train-ocrnet» из репозитория: # https://github.com/nvidia/skills
2Положить папку с SKILL.md к навыкам ассистента
# положите папку навыка туда, где ассистент ищет skills: cp -r tao-train-ocrnet/ ~/.claude/skills/tao-train-ocrnet/

Навык — папка с файлом SKILL.md (описание + инструкции) и опциональными скриптами. Ассистент (Claude, Claude Code и совместимые) подхватывает его по описанию и следует шагам.