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

TAO обучение ReID

Person re-identification (ReID). Learns discriminative embeddings to match the same person across different camera views, based on metric learning. Use when training, evaluating, exporting, or running inference for a TAO person re-identification model. Trigger phrases include

nvidia official Другое

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

Навык обучает модели повторной идентификации людей (ReID) на основе метрического обучения для сопоставления одних и тех же людей на разных ракурсах камер. Он поддерживает полный цикл работы: обучение, оценку, инференс и экспорт моделей TAO в формате ONNX.

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

  • Обучение различающих эмбеддингов для распознавания людей в системах видеонаблюдения.
  • Оценка качества модели по метрикам mAP и CMC (Cumulative Matching Characteristic) на тестовых и запросовых наборах данных.
  • Инференс для поиска совпадений человека в галерее изображений по запросу.
  • Экспорт обученной модели в формат ONNX для последующего развертывания.

Как работает

  1. Инициализировать окружение через навык tao-setup, если сессия не была запущена плагином банка навыков TAO.
  2. Подготовить спецификацию (spec.yaml) с указанием путей к данным (train, test, query datasets) и чекпоинтам, используя обязательные ключи из таблицы требований к данным.
  3. Запустить контейнер Docker с образом nvcr.io/nvidia/tao/tao-toolkit:7.2.0-pyt, монтируя локальные папки data, specs и results.
  4. Выполнить нужное действие (train, evaluate, inference, export), передав путь к спецификации через флаг -e.
  5. При использовании AutoML для обучения убедиться, что оптимизируется метрика cmc_rank_1 (maximize), а не val_loss.

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

Обучи модель ReID на моих данных, используя AutoML для оптимизации метрики cmc_rank_1.
Запусти оценку модели ReID по чекпоинту из предыдущего запуска и покажи метрики mAP и CMC.
Экспортируй обученную модель ReID в ONNX файл для инференса.

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

  • Доступ к Docker и GPU (минимум 1 GPU с 16GB+ VRAM).
  • Установленный навык tao-setup для предварительной настройки окружения.
  • Наборы данных в формате tar.gz (sample_train, sample_test, sample_query) с правильной структурой папок идентификаторов.

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

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

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

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