数据与分析 ✓ NVIDIA · 官方 medical-imagingmriniftidiffusion-modelfine-tuningmonaicuda

NV 脑部 MRI 扩散模型微调

从 NIfTI 脑部 MRI 数据微调 NV-Generate-CTMR 的扩散 UNet。

FollowSkills 评估 · FSRS-2.0
不推荐
51/ 100 五分制 2.6 / 5
信任安全16 / 25 · 3.2/5

明确限制为工程研发用途,说明输出、网络端点、缓存、GPU训练及目录写入,并记录上游提交;但允许调用Bash、安装依赖、访问GitHub/Hugging Face、修改用户Python环境和写入家目录,缺少逐次用户确认、敏感MRI数据处理说明、权限隔离与回滚机制,因此扣分。

可靠稳定8 / 20 · 2.0/5

入口脚本、配置分层、输入校验、阶段化命令、退出码和stderr尾部反馈较清楚,且有单元测试文件;但未提供可核实的运行结果或CI覆盖,真实路径依赖上游仓库、CUDA、权重和依赖版本,部分异常场景和失败后的恢复不足,按静态上限保守评分。

适用触发8 / 15 · 2.7/5

目标用户、NIfTI/MONAI datalist格式、模态选项、预检与训练边界说明较明确,也明确非临床用途;但仅面向具备CUDA和上游环境的研发场景,未说明中文交互支持,且核心依赖GitHub/Hugging Face,面向中国大陆网络环境的可达性和替代方案未交代,因此扣分。

规范维护9 / 15 · 3.0/5

SKILL.md包含目的、前置条件、命令、配置、监控、调参、限制和故障排查;manifest提供版本、输入输出、依赖、上游固定提交和副作用,Apache-2.0及作者信息明确。但缺少正式变更记录、长期维护责任和更新流程,skill-card中的版本信息与评审修订存在不一致,目录规范检查还报告作者格式问题。

有效结果6 / 15 · 2.0/5

包装器覆盖数据预检、配置暂存、嵌入生成、训练、可选推理和结果摘要,预检优先流程对核心任务有直接帮助;但实际训练与生成依赖外部仓库、模型权重、CUDA和用户数据,默认两轮训练仅用于验证,且没有静态可验证的代表性产物或质量评估,因此不能给出高于静态上限的分数。

证据核验4 / 10 · 2.0/5

代码、manifest、测试、评测报告和固定上游提交提供了一定审计线索,输出还记录命令、提交和stderr;但本次只能静态阅读,评测报告仅含小样本、无负向任务,未提供可独立复现的CI证据或真实训练结果,因此证据覆盖有限。

证据充分度: 评估于 2026年7月20日 审查版本 55f18499943e
上游仓库在本次评估后已有新提交;当前评分仍对应所示审查版本,可能尚未覆盖最新改动。
使用前请注意
  • 执行前应在隔离环境中明确确认Bash、依赖安装、网络访问、家目录缓存和输出目录写入范围。
  • 不要把生成的MRI或检查点用于临床、监管、生产训练数据批准或未经专家审查的模型结论。
  • 应核验上游仓库、模型权重、依赖版本及网络可达性,并为敏感影像数据制定本地化存储和日志脱敏策略。
查看完整评分方法 →

这个 Skill 能做什么,适合哪些场景?

该技能用于使用用户提供的 NIfTI 训练体数据微调 NV-Generate-CTMR 的 rflow-mr-brain 扩散 UNet。它通过 Python 包装脚本准备配置、生成潜变量嵌入并调用上游训练脚本,也可执行可选推理。推荐先运行无 GPU 训练的 preflight 检查。该技能不适用于临床解读、监管用途、自动诊断或批准合成数据用于生产训练。

读取 MONAI 风格 datalist JSON 和数据根目录,校验相对图像路径及模态信息;运行 scripts/run_mr_brain_finetune.py;暂存配置和 datalist,调用 scripts.diff_model_create_training_data 生成嵌入,写入嵌入 sidecar,并调用 scripts.diff_model_train;可选调用 scripts.diff_model_infer。它在输出目录生成嵌入、检查点、日志、工作流摘要和可选推理图像,并将结果 JSON 输出到标准输出。运行时可能从 Hugging Face 和 GitHub 获取模型资产或上游代码。

  1. 医疗 AI 研究人员希望用自有 NIfTI 脑部 MRI 训练体快速验证微调流程时。
  2. 模型开发者需要针对 mri_t1、mri_t2、mri_flair 等支持的脑部 MRI 模态调整扩散 UNet 时。
  3. 工程师希望在正式 GPU 训练前检查 datalist、数据路径和配置是否完整时。
  4. 研究团队需要生成微调检查点并通过内置推理或 nv-generate-mr-brain 技能进行定性检查时。

这个 Skill 有哪些优点和局限?

优点
  • 提供从 datalist 校验、嵌入生成到检查点输出的完整包装流程。
  • 支持 preflight、混合精度、多 GPU 和可选推理。
  • 支持通过配置 JSON 调整学习率、批大小、缓存率和推理参数。
  • 记录检查点、嵌入 sidecar、推理输出及命令结果,便于运行追踪。
局限
  • 真实训练需要 NVIDIA CUDA GPU、上游依赖和下载的 MR-brain 权重。
  • 首次运行会预计算潜变量嵌入,训练成本和启动时间可能较高。
  • 训练结果可能因硬件、CUDA 和软件包版本而不具确定性。
  • 技能只管理文件记录和命令来源,不验证解剖真实性或下游模型效用,也没有提供临床或生产批准依据。

如何安装这个 Skill?

安装 NVIDIA 技能集合:npx skills add nvidia/skills --skill nv-generate-mr-brain-finetune --yes。README 未说明该单个技能的独立手工安装步骤;运行时还需要从 Medical AI Skills 仓库根目录执行脚本,并准备 NV-Generate-CTMR 上游 checkout。

如何使用这个 Skill?

先运行 preflight:export NV_GENERATE_ROOT="${NV_GENERATE_ROOT:-$HOME/.cache/nvidia-skills/upstreams/NV-Generate-CTMR-61c4ec7}" && python skills/nv-generate-mr-brain-finetune/scripts/run_mr_brain_finetune.py PATH_TO_DATALIST.json --data-base-dir PATH_TO_DATA_ROOT --output-dir runs/nv_generate_mr_brain_finetune_preflight --preflight。确认输入有效后,安装上游 requirements.txt,再移除 --preflight 并按需加入 --epochs、--modality、--num-gpus 或 --run-inference。不要使用 fixture datalist 进行真实训练。

常见问题

可以直接用于临床或生产训练数据审批吗?
不可以。源文档明确排除临床解读、监管用途、自动诊断、临床部署和生产训练数据批准。
运行 preflight 需要 GPU 吗?
推荐的 preflight 用于检查和暂存,不启动 GPU 训练;真实微调需要 NVIDIA CUDA GPU。
为什么会出现找不到 diffusion training scripts?
通常是 NV_GENERATE_ROOT 未指向包含 scripts/diff_model_create_training_data.py、diff_model_train.py 和 diff_model_infer.py 的当前 NV-Generate-CTMR checkout。
技能会访问网络或写入本地文件吗?
会。它在输出目录写入配置、嵌入、检查点和日志,并可能访问 Hugging Face 获取模型资产、访问 GitHub 获取上游 checkout。

同仓库的其他 Skills

均来自 NVIDIA/skills

数据与分析 ✓ NVIDIA · 官方

NV-Segment-CT 微调

为 CT NIfTI 标注执行 NV-Segment-CT VISTA3D 微调。

数据与分析 ✓ NVIDIA · 官方

NV-Generate-MR

通过 NVIDIA 的 rflow-mr 流程生成合成体部 MRI 影像体积。

数据与分析 ✓ NVIDIA · 官方

NV-Segment-CTMR 医学影像分割

在 CT 或 MRI NIfTI 体数据上运行分割并生成可核查证据。

数据与分析 ✓ NVIDIA · 官方

脑部 MRI 合成生成

通过 NVIDIA 工作流生成合成脑部 MRI 体积。

数据与分析 ✓ NVIDIA · 官方

NV-Generate-CT 合成 CT 生成

通过 NVIDIA 的 rflow-ct 流程生成带配对掩码的合成 CT 体数据。

数据与分析 ✓ NVIDIA · 官方

NV-Generate 医学影像 VAE 微调

为 CT/MRI NIfTI 训练体数据微调 NV-Generate-CTMR MAISI VAE。

数据与分析 ✓ NVIDIA · 官方

NV-Segment-CT 医学影像分割技能

在 CT NIfTI 体数据上运行 VISTA3D 分割,并生成可核验的标签图证据。

数据与分析 ✓ NVIDIA · 官方

DICOM CT 到 NIfTI 体数据转换

将单个 CT DICOM 序列转换为带仿射证据的 HU NIfTI 体数据。

开发与工程 ✓ NVIDIA · 官方

i4h 导管导航工作流概览

帮助智能体理解并选择血管内导管导航工作流的正确阶段。

开发与工程 ✓ NVIDIA · 官方

导管导航 DRR 渲染

从 CT 缓存或合成体模生成单帧数字重建放射图,用于荧光透视预览与渲染冒烟测试。

数据与分析 ✓ NVIDIA · 官方

Nemotron 模型定制流水线

为 Nemotron 模型规划、配置并串联数据处理、训练、对齐、转换、优化与评测步骤。

开发与工程 ✓ NVIDIA · 官方

NeMo MBridge 配方推荐器

根据模型、GPU规模和训练目标推荐可调整的Megatron Bridge配方。

数据与分析 ✓ NVIDIA · 官方

Nemotron 检索配方助手

帮助规划、调试、微调、评估和部署 Nemotron 嵌入与重排检索流程。

数据与分析 ✓ NVIDIA · 官方

DICOM 序列预检

在转换或推理前快速验证单个 DICOM 序列目录。

数据与分析 ✓ NVIDIA · 官方

Cosmos-Embed 视频检索与微调技能

用 Cosmos-Embed1 将视频与文本嵌入用于检索和微调。

数据与分析 ✓ NVIDIA · 官方

DICOM 元数据提取器

从单个 DICOM 文件提取选定元数据并标记标准标签中的 PHI。

开发与工程 ✓ NVIDIA · 官方

NV-Reason-CXR 胸部X光推理测试技能

为胸部X光模型执行可复现的命令形状与在线推理冒烟测试。

数据与分析 ✓ NVIDIA · 官方

临床语音识别微调

用 NeMo 对 Parakeet TDT v2 进行临床词汇微调,并以离线 KER 评估闭环效果。

数据与分析 ✓ NVIDIA · 官方

导管导航血管数字孪生

将CT血管成像处理并分割为可用于导管导航的患者特异性血管数字孪生。

数据与分析 ✓ NVIDIA · 官方

PAIDF AnomalyGen 异常图像生成

微调模型并生成、评估和筛选合成异常图像。

相关 Skills