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

TAO обучение OCDNet

OCDNet for scene text detection. Detects arbitrary-oriented text regions in natural images using a differentiable binarization approach. Use when training, evaluating, exporting, pruning, quantizing, retraining, or running inference for a TAO OCDNet model. Trigger phrases include

nvidia official Дизайн

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

OCDNet — модель для детекции текстовых областей произвольной ориентации на натуральных изображениях с использованием метода дифференцируемой бинаризации. Позволяет обучать, оценивать, экспортировать, прунить, квантовать, дообучать или выполнять инференс модели через TAO OCDNet CLI.

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

  • Обучение модели на новых данных с поддержкой AutoML по умолчанию.
  • Выполнение инференса или оценки качества детекции текста.
  • Прунинг и квантование модели для оптимизации ресурсов.
  • Дообучение (retrain) модели с использованием ранее пруненного графа.
  • Генерация TensorRT-движка для развертывания через TAO Deploy.

Как работает

  1. Убедитесь, что сессия инициализирована плагином TAO skill bank; если нет, запустите навык tao-setup.
  2. При запросе обучения проверьте references/skill_info.yaml для определения политики AutoML (по умолчанию automl_policy: on).
  3. Если AutoML включен и доступны схемы, направьте действие train через tao-skill-bank:tao-run-automl; иначе используйте прямой вызов ocdnet train.
  4. Для дообучения используйте команду ocdnet train с параметрами model.load_pruned_graph: true и model.pruned_graph_path.
  5. Для TensorRT-действий (генерация движка, оценка, инференс) сначала прочитайте references/tao-deploy-ocdnet.md.
  6. Используйте файлы из schemas/ и references/ для получения параметров, границ и шаблонов спецификаций.

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

Обучи OCDNet на моих данных с включенным AutoML.
Выполни инференс модели OCDNet для детекции текста на изображении.
Проведи прунинг и квантование модели OCDNet.

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

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

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

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

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

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