AIKraft/AI/Войти
AIKraft/Skills/TAO обучение Depth Anything V2

TAO обучение Depth Anything V2

Monocular depth estimation using Metric Depth Anything v2 or Relative Depth Anything architectures. Predicts per-pixel depth from single RGB images. Use when training, evaluating, exporting, or running inference for a TAO monocular depth model. Trigger phrases include

nvidia official Дизайн

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

Навык обеспечивает одноканальную оценку глубины с использованием архитектур Metric Depth Anything v2 или Relative Depth Anything, предсказывая поточечную глубину по одиночным RGB-изображениям. Он поддерживает действия обучения, оценки, экспорта, квантования и инференса через унифицированный CLI depth_net в контейнере TAO. Для действий TAO Deploy (генерация TensorRT, оценка и инференс TensorRT) требуется предварительное чтение документации и использование специфичного контейнера.

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

  • Обучение модели глубины с использованием AutoML (по умолчанию) или без него (plain training).
  • Инференс и оценка производительности модели на тестовых данных.
  • Экспорт обученной модели в формате, пригодном для развертывания.
  • Квантование модели для оптимизации ресурсов.
  • Генерация TensorRT-движка и инференс с использованием TAO Deploy.

Как работает

  1. Убедиться, что сессия инициализирована плагином TAO skill bank; если нет, выполнить навык tao-setup.
  2. Определить тип модели (mono/stereo) через параметр model.model_type и загрузить соответствующий предобученный чекпоинт согласно матрице в references/parameters.md.
  3. Для действий обучения проверить наличие схем и шаблонов; если automl_policy включен, направить запрос через tao-skill-bank:tao-run-automl, иначе использовать прямое обучение.
  4. Для действий оценки, инференса и экспорта использовать соответствующие подкоманды CLI depth_net внутри контейнера.
  5. Для генерации TensorRT-движка (gen_trt_engine) использовать контейнер TAO Deploy, так как PyT-версия не поддерживает это действие.
  6. Сохранять переопределения пользовательского рабочего процесса для датасетов, спецификаций, директорий вывода и настроек GPU.

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

Обучи модель Metric Depth Anything v2 с включенным AutoML.
Запусти инференс одноканальной оценки глубины на тестовом изображении.
Экспортируй модель глубины для последующего развертывания.

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

  • Инициализация сессии через навык tao-setup (preflight, credentials, cross-skill discovery).
  • Контейнер TAO для действий train, evaluate, inference, export, quantize.
  • Контейнер TAO Deploy для действий gen_trt_engine, TensorRT evaluate и TensorRT inference.

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

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

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

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