AIKraft/AI/Войти
AIKraft/Skills/nv: сегментация ct (дообучение)

nv: сегментация ct (дообучение)

Used for smoke or dataset finetuning of NV-Segment-CT VISTA3D on CT NIfTI labels. Not for clinical validation.

nvidia official Данные

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

Навык предназначен для smoke-тестирования или дообучения (finetuning) модели NV-Segment-CT VISTA3D на датасетах с CT-метками NIfTI, включая фиксированный канал softmax. Он не предназначен для клинической валидации и оборачивает встроенный скрипт MONAI bundle.

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

  • Проверка работоспособности конвейера (smoke-test) с использованием микро-датасета.
  • Воспроизведение результатов для MSD Task06 (Lung Tumor) в режиме sanity.
  • Дообучение модели на пользовательских данных с автоматической сегментацией.
  • Дообучение с фиксированным каналом softmax для взаимно исключающих классов.
  • Отслеживание метрик обучения через MLflow.

Как работает

  1. Установить зависимости (monai==1.4.0, numpy<2 и др.) и подготовить окружение.
  2. При использовании --softmax выполнить настройку исходного кода NV-Segment-CTMR.
  3. Запустить скрипт scripts/run_finetune.py из корня репозитория с нужными аргументами.
  4. Указать параметры датасета (--dataset-dir, --datalist), анатомию и маппинг меток.
  5. Выбрать режим (--smoke, --sanity, --auto-seg или --softmax) и каталог вывода.
  6. Проверить результат в output.json, обращая внимание на recommended_ckpt и метрики Dice.

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

Запусти smoke-тест на микро-датасете с размером патча 64x64x64.
Выполни sanity-репродукцию для MSD Task06 (Lung Tumor) с маппингом меток [[1, 23]].
Обучи модель на моих данных из /path/to/dataset с маппингом [[1,3],[2,13]] в режиме softmax.

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

  • Python 3.10+ с CUDA-совместимым Torch и пакетом monai==1.4.0.
  • Доступ к интернету для загрузки моделей с Hugging Face и исходников с GitHub.
  • Опционально: MLflow >=2.10,<4 для отслеживания экспериментов.

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

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

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

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