AIKraft/AI/Войти
AIKraft/Skills/TAO обучение распознаванию действий

TAO обучение распознаванию действий

Action recognition from video sequences. Supports RGB, optical flow, and joint (multi-stream) input types for classifying temporal actions in video clips. Use when training, evaluating, exporting, or running inference on a TAO action-recognition model. Trigger phrases include

nvidia official Другое

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

Навык предназначен для обучения, оценки, экспорта и инференса модели распознавания действий (action recognition) по видеопоследовательностям. Он поддерживает входные типы RGB, оптический поток и совместный (multi-stream) для классификации временных действий в видеоклипах.

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

  • Обучение модели классификации действий на пользовательских видео-датасетах с использованием AutoML или прямого обучения.
  • Оценка точности обученной модели на тестовых данных для получения метрик accuracy и m_accuracy.
  • Инференс (прогноз) действий на новых видеоклипах с использованием сохраненных весов модели.
  • Экспорт обученной модели в формат ONNX для последующего развертывания.

Как работает

  1. Подготовить данные: извлечь архивы (train.tar.gz, test.tar.gz) в директории и указать пути в spec_overrides.
  2. Настроить параметры через YAML-спецификацию (например, dataset.label_map, model.model_type, train.num_epochs) или использовать AutoML.
  3. Запустить контейнер Docker с образом nvcr.io/nvidia/tao/tao-toolkit, монтируя директории данных, спецификаций и результатов.
  4. Выполнить требуемое действие (train, evaluate, inference, export), передав путь к файлу спецификации через флаг -e.
  5. Для инференса и экспорта указать путь к контрольной точке (checkpoint), полученной на этапе обучения.

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

Обучи модель распознавания действий на моих видео, используя AutoML.
Запусти инференс модели action_recognition на тестовых клипах и экспортируй результат в ONNX.
Оцени точность модели на тестовом датасете, указав путь к контрольной точке.

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

  • Docker с поддержкой GPU (NVIDIA) и установленный образ nvcr.io/nvidia/tao/tao-toolkit.
  • Доступ к TAO SDK или плагину skill bank для инициализации и диспетчеризации задач.
  • Подготовленные видео-датасеты в виде извлеченных директорий (не архивов).

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

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

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

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