用途:调用 NVIDIA-Medtech/NV-Generate-CTMR 的 rflow-mr-brain 工作流,生成仅含图像的合成脑部 MRI 体积(NIfTI)。封装本身不重新实现扩散采样或自编码器解码,只负责配置覆盖暂存、输出路径与校验。
主要能力:输入为模型配置覆盖(model_config_override),输出为 synthetic_mr_brain_volumes 与 result_json。支持的模态名包括 mri、mri_t1、mri_t2、mri_flair、mri_swi、mri_t1_skull_stripped、mri_t2_skull_stripped、mri_flair_skull_stripped、mri_swi_skull_stripped。fixture 是可传入的 JSON 覆盖,常用键有 dim、spacing、num_inference_steps、cfg_guidance_scale、modality;传 default 表示使用上游默认值。每次运行记录暂存配置、模型清单、上游命令、输出几何、间距、仿射、强度范围及有限/非常量数据检查。
使用方式:先按需克隆并安装上游仓库、下载 rflow-mr-brain 权重,然后运行封装脚本:
export NV_GENERATE_ROOT="${NV_GENERATE_ROOT:-.workbench_data/upstreams/NV-Generate-CTMR}" &&
python -m pip install -r "$NV_GENERATE_ROOT/requirements.txt" &&
python skills/nv-generate-mr-brain/scripts/run_mr_brain.py PATH_TO_MR_BRAIN_CONFIG.json --output-dir OUT_DIR --modality mri_t1 --random-seed 1234
若宿主代理提供 run_script,则用 run_script("scripts/run_mr_brain.py", args=[...]) 调用。不要自行编写推理代码,不要添加 rm/mkdir 或清理 --output-dir,封装会创建该目录;应使用全新的 --output-dir。
限制:仅生成图像,无分割掩码;输出为合成数据,未经独立质量复审不得作为生产医疗模型训练数据。不用于临床部署、临床解读、自主诊断或监管申报。需要至少 16 GB 显存的 NVIDIA GPU,上游路径无 CPU 回退。依赖漂移可能导致导入错误;输入路径错误或模态不支持可能产生空或不符合 schema 的输出。修改参数、副作用或校验门之前需阅读 skill_manifest.yaml。