AIKraft/Skills/nv: генерация mr мозга (дообучение)

nv: генерация mr мозга (дообучение)

Used for finetuning NV-Generate-CTMR MR-brain diffusion UNet from a NIfTI datalist. Not for clinical or production data approval.

nvidia official Данные

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

Навык предназначен для дообучения (finetuning) диффузионной UNet NV-Generate-CTMR rflow-mr-brain v1 на основе пользовательских объемов данных NIfTI (T1, T2, FLAIR, SWI или MRA). Он не предназначен для клинической интерпретации, регуляторного использования или одобрения синтетических данных для промышленного обучения.

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

  • Адаптация модели NV-Generate-CTMR под специфические наборы медицинских изображений мозга, предоставленные пользователем.
  • Подготовка и запуск процесса обучения диффузионной модели с использованием локально размеченных данных NIfTI.
  • Генерация синтетических данных MR-изображений мозга после завершения процесса дообучения модели.

Как работает

  1. Агент принимает на вход список данных (datalist) и базовую директорию с данными (data_base_dir).
  2. Локально подготавливается конфигурационный файл (config glue) на основе контракта upstream (JSON, аналогичный cell [10] из train_diff_unet_tutorial.ipynb).
  3. Выполняются скрипты upstream: scripts.diff_model_create_training_data для подготовки данных и scripts.diff_model_train для обучения.
  4. Опционально запускается скрипт scripts.diff_model_infer для получения результатов инференса.
  5. В качестве результата возвращаются дообученный чекпоинт (finetuned_checkpoint), опциональные выводы инференса (inference_outputs) и файл result_json.

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

Дообучи модель NV-Generate-CTMR на моих данных T1 NIfTI из папки /data/brain.
Запусти процесс finetuning диффузионной UNet для генерации MR-изображений мозга с использованием предоставленного списка NIfTI файлов.
Проведи дообучение модели rflow-mr-brain v1 и получи чекпоинт, используя данные FLAIR и SWI.

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

  • Доступ к скриптам upstream: scripts.diff_model_create_training_data, scripts.diff_model_train, scripts.diff_model_infer.
  • Наличие данных обучения в формате NIfTI (T1, T2, FLAIR, SWI или MRA).

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

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

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

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