# FreeSurfer synth-tools Python env — AMD ROCm 6.0 profile.
# torch wheel picked from --extra-index-url https://download.pytorch.org/whl/rocm6.0
# Note: TensorFlow does NOT ship a ROCm GPU wheel; the plain tensorflow wheel
# runs on the CPU, so SynthSeg/SynthSR still work on AMD systems (CPU inference).
# Only SynthStrip (PyTorch-based) benefits from ROCm acceleration here.

tensorflow>=2.19,<2.20
# Keras 2 API for the synth tools; same minor as tensorflow (see requirements-cpu.txt).
tf-keras>=2.19,<2.20
torch>=2.2,<2.6

surfa>=0.5
nibabel>=5.1
numpy>=1.24,<3
scipy>=1.10
h5py>=3.10
scikit-image>=0.21
scikit-learn>=1.3
SimpleITK>=2.3

# --- SynthMorph ------------------------------------------------------------
# mri_synthmorph imports voxelmorph and neurite, which are not on PyPI at the
# versions FreeSurfer uses; these are upstream's commits
# (freesurfer-8.2.0/python/requirements-build.txt). Needs git at setup time.
neurite @ git+https://github.com/adalca/neurite.git@95b2b568b124cbc654467177ddcdb2cb3526788c
voxelmorph @ git+https://github.com/voxelmorph/voxelmorph.git@feb74e0541b8a390ccd2ea57b745aa8808703ca4

pyyaml
tqdm
matplotlib
