AIKraft/AI/Войти
AIKraft/Skills/Импорт модели компьютерного зрения DeepStream

Импорт модели компьютерного зрения DeepStream

Use this skill to bring any vision model from HuggingFace or NVIDIA NGC into an NVIDIA DeepStream pipeline with end-to-end automation: ONNX download, SafeTensors export, TRT engine build, custom nvinfer bbox parser, multi-stream benchmark, and PDF report. Object detection models only.

nvidia official Документы

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

Автоматизирует перенос моделей детекции объектов из HuggingFace или NVIDIA NGC в пайплайн NVIDIA DeepStream, включая конвертацию в ONNX, сборку TensorRT-движка и запуск бенчмарков. Генерирует детальный PDF-отчёт с графиками производительности для валидации работы модели.

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

  • Миграция существующей модели детекции (например, RT-DETR) в среду NVIDIA DeepStream для видеостриминга.
  • Автоматизированное тестирование производительности модели на разных батч-размерах и потоках.
  • Получение стандартизированного отчёта с метриками (FPS, задержки) в формате PDF/HTML.
  • Валидация качества детекции на тестовых видео с сохранением кадров и результатов в формате KITTI.

Как работает

  1. Предложить пользователю выбор: использовать рекомендованную модель PekingU/rtdetr_r50vd или указать свой источник (HuggingFace/NGC).
  2. Выполнить предполётную проверку (preflight) через Docker-контейнер DeepStream для подтверждения доступности GPU и инструментов.
  3. Загрузить модель, экспортировать её в формат ONNX (или SafeTensors) и собрать динамический TensorRT-движок (TRT engine).
  4. Запустить бенчмарки в DeepStream: одиночный поток, многопоточное тестирование и дамп детекций в формате KITTI.
  5. Сгенерировать отчёт с 5 графиками, данными в JSON и финальными файлами HTML и PDF.

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

Запусти полный пайплайн для модели PekingU/rtdetr_r50vd.
Импортируй модель с HuggingFace в DeepStream и покажи бенчмарк.
Проверь совместимость модели NGC с DeepStream и сгенерируй отчёт.

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

  • Docker с поддержкой NVIDIA GPU (драйверы NVIDIA и WSL2 для Windows).
  • Доступ к интернету для загрузки моделей с HuggingFace или NVIDIA NGC.
  • Образ nvcr.io/nvidia/deepstream:9.1-triton-multiarch.

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

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

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

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