⚡

NV-Generate-MRSkill nv-generate-mr

该技能用于基于 NVIDIA NV-Generate-CTMR 的 rflow-mr 扩散模型生成合成的人体 MRI（磁共振）体积图像，支持 mri_t1/mri_t2/mri_flair 多种模态。它封装了上游推理流程，输出 NIfTI 格式的合成影像，不与分割标签配对，适用于医疗影像生成、合成数据扩充和算法验证等场景。关键词：MRI生成、医学影像合成、扩散模型、NVIDIA GenAI、rflow-mr、体数据生成。

医学影像生成 0 次安装 1 次浏览 更新于 9/6/2026
名称 nv-generate-mr
描述 用于通过NV-Generate-CTMR rflow-mr生成合成的人体MRI体积。不用于配对掩膜或生产训练数据。
开源协议 Apache-2.0 allowed-tools: Bash metadata:
作者 NVIDIA MedTech Team tags: - MedTech - MRI - generation

NV-Generate-MR

目的

  • 使用NV-Generate-CTMR rflow-mr生成合成的人体MRI体积，不用于配对掩膜或生产训练数据。
  • 严格按照文档使用包装脚本，避免替换上游入口或自定义推理代码。
  • 清单输入为 model_config_override，输出为 synthetic_mr_volumes 和 result_json。

说明

  • 修改前需阅读 skill_manifest.yaml。
  • 通过 scripts/run_mr.py 运行，输出到指定目录。
  • 在同一个 Bash 命令中安装 requirements，防止全新环境缺少 nibabel/MONAI。
  • 不要清理 --output-dir；wrapper 会自动创建。
  • 运行结果需结合 JSON 输出及验证器规则判断。

可用脚本

  • scripts/run_mr.py：主要入口，参数为 MODEL_CONFIG.json --output-dir OUT_DIR --modality mri_t1 [–random-seed N] [–yes]

先决条件

  • 需要 GPU/CUDA 和清单中声明的 Python 包。
  • 可能写入 ~/.cache/huggingface/，并访问 huggingface.co 或 github.com。

限制

  • 基于 NVIDIA-Medtech/NV-Generate-CTMR 的扩散推理，不支持分割掩膜生成。
  • 脑部 MRI 建议使用 rflow-mr-brain，进入技能 nv-generate-mr-brain。
  • 权重按 NVIDIA 非商业协议使用，输出不得直接用于生产训练。
  • 不适用于临床、诊断、监管等场景。

故障排除

  • 依赖缺失：安装清单对应包。
  • 输出为空：检查输入路径和模态，重新运行测试。
  • 验证门失败：保留证据，按提示修复。

详细运行命令请参考原始技能文档。