Add official VoxelMorph CT registration pipeline

This commit is contained in:
admin
2026-06-03 00:30:32 +08:00
parent e8d8f2c468
commit 2dba05ae4a
16 changed files with 2130 additions and 0 deletions

23
scripts/setup_env.sh Executable file
View File

@@ -0,0 +1,23 @@
#!/usr/bin/env bash
set -euo pipefail
ENV_NAME="${ENV_NAME:-voxelmorph-head-ct}"
conda env create -f environment.yml
conda run -n "${ENV_NAME}" python -m pip install --no-cache-dir -r requirements.txt
conda run -n "${ENV_NAME}" python - <<'PY'
import os
os.environ.setdefault("NEURITE_BACKEND", "pytorch")
os.environ.setdefault("VXM_BACKEND", "pytorch")
import torch
import voxelmorph as vxm
import neurite as ne
print("torch:", torch.__version__)
print("cuda:", torch.cuda.is_available())
if torch.cuda.is_available():
print("gpu:", torch.cuda.get_device_name(0))
print("voxelmorph VxmPairwise:", hasattr(vxm.nn.models, "VxmPairwise"))
print("neurite NCC:", hasattr(ne.nn.modules, "NCC"))
PY