如何创建Dockerfile遍历文件夹文件执行mri_synthstrip命令

背景 #

mri_synthstrip是一款脑MRI预处理工具，可输入脑MRI文件并输出去除颅骨后的文件：

• 本地单文件处理命令：mri_synthstrip -i input.nii.gz -o stripped.nii.gz
• 使用Docker镜像freesurfer/synthstrip的单文件处理命令：

  docker run -v C:/Users/ch134560/Desktop/MRI_files://data freesurfer/synthstrip -i //data/input.nii.gz -o //data/stripped.nii.gz
  

问题 #

我希望创建Docker镜像，实现遍历目录中所有MRI文件并调用synthstrip批量去颅骨，但遇到以下问题：

• 编写的多阶段Dockerfile中，循环内的freesurfer/synthstrip -i "$file" -o "$file"或mri_synthstrip -i "$file" -o "$file"命令无法被识别。
• 尝试过通过Python调用该命令的多阶段构建方案，但仍不清楚如何正确调用synthstrip命令。
• 考虑过创建轻量Docker镜像批量调用synthstrip镜像，但不知具体实现方法，特此求助。

解决方案 #

方法1：基于官方镜像构建批量处理镜像 #

直接以freesurfer/synthstrip为基础镜像，添加批量处理脚本，避免多阶段构建导致的命令找不到问题：

Dockerfile

FROM freesurfer/synthstrip

# 添加批量处理脚本
COPY run_batch.sh /usr/local/bin/
RUN chmod +x /usr/local/bin/run_batch.sh

# 设置工作目录为挂载的数据目录
WORKDIR /data

# 容器启动时自动执行批量脚本
ENTRYPOINT ["run_batch.sh"]

配套run_batch.sh脚本（与Dockerfile同目录）

#!/bin/bash

# 遍历当前目录下所有.nii.gz格式的MRI文件
for file in *.nii.gz; do
  if [ -f "$file" ]; then
    # 生成带后缀的输出文件名，避免覆盖原文件
    output="${file%.nii.gz}_stripped.nii.gz"
    # 直接调用镜像内置的mri_synthstrip命令
    mri_synthstrip -i "$file" -o "$output"
    echo "处理完成：$file -> $output"
  fi
done

使用步骤

1. 将Dockerfile和run_batch.sh放在同一文件夹
2. 构建镜像：docker build -t synthstrip-batch .
3. 运行容器并挂载本地MRI文件目录：

   # Linux/macOS
   docker run -v /your/local/mri/path:/data synthstrip-batch
   
   # Windows
   docker run -v C:/your/local/mri/path:/data synthstrip-batch
   

方法2：轻量镜像批量调用官方synthstrip容器 #

如果不想基于官方镜像构建，可使用轻量的Alpine镜像，通过Docker-in-Docker方式批量调用官方容器（需宿主已安装Docker）：

Dockerfile

FROM alpine:latest

# 安装Docker客户端（与宿主Docker版本兼容）
RUN apk add --no-cache docker-cli

# 添加批量处理脚本
COPY run_batch_dind.sh /usr/local/bin/
RUN chmod +x /usr/local/bin/run_batch_dind.sh

WORKDIR /data

ENTRYPOINT ["run_batch_dind.sh"]

配套run_batch_dind.sh脚本

#!/bin/bash

# 遍历目录下的MRI文件
for file in *.nii.gz; do
  if [ -f "$file" ]; then
    output="${file%.nii.gz}_stripped.nii.gz"
    # 调用官方synthstrip容器处理单个文件，处理完成后自动删除容器
    docker run --rm -v "$(pwd)":/data freesurfer/synthstrip -i "/data/$file" -o "/data/$output"
    echo "处理完成：$file -> $output"
  fi
done

使用步骤

1. 构建镜像：docker build -t synthstrip-batch-light .
2. 运行时需挂载宿主Docker socket，让容器内的Docker客户端能调用宿主服务：

   # Linux/macOS
   docker run -v /your/local/mri/path:/data -v /var/run/docker.sock:/var/run/docker.sock synthstrip-batch-light
   
   # Windows
   docker run -v C:/your/local/mri/path:/data -v //var/run/docker.sock:/var/run/docker.sock synthstrip-batch-light
   

方法3：Python脚本调用的多阶段构建方案 #

通过多阶段构建将Python环境与synthstrip工具结合，用Python实现批量逻辑：

Dockerfile

# 第一阶段：获取synthstrip工具及依赖
FROM freesurfer/synthstrip as synthstrip-base

# 第二阶段：构建Python运行环境
FROM python:3.9-slim

# 从第一阶段复制synthstrip工具到当前镜像
COPY --from=synthstrip-base /opt/freesurfer /opt/freesurfer
# 添加工具路径到系统环境变量
ENV PATH="/opt/freesurfer/bin:${PATH}"

# 添加Python批量处理脚本
COPY batch_process.py /usr/local/bin/
RUN chmod +x /usr/local/bin/batch_process.py

WORKDIR /data

# 启动容器时执行Python脚本
ENTRYPOINT ["python", "/usr/local/bin/batch_process.py"]

配套batch_process.py脚本

import os
import subprocess

# 获取当前目录下所有.nii.gz格式的MRI文件
mri_files = [f for f in os.listdir('.') if f.endswith('.nii.gz') and os.path.isfile(f)]

for file in mri_files:
    output_file = file.replace('.nii.gz', '_stripped.nii.gz')
    # 调用mri_synthstrip命令
    cmd = ['mri_synthstrip', '-i', file, '-o', output_file]
    try:
        subprocess.run(cmd, check=True)
        print(f"处理完成：{file} -> {output_file}")
    except subprocess.CalledProcessError as e:
        print(f"处理失败 {file}: {str(e)}")

使用步骤

与方法1一致，构建镜像后挂载本地MRI目录运行即可。

内容的提问来源于stack exchange，提问作者Ivan