English
medical
brain-data
mri
brain2vec-v2 / requirements.txt
jesseab's picture
Add Brain2Vec-v2 files and model card
0556c0e
# PyTorch (CUDA or CPU version).
torch>=1.12
# Install MONAI Generative first
monai-generative
# Install the latest MONAI directly from GitHub (development version)
git+https://github.com/Project-MONAI/MONAI.git#egg=monai
# For perceptual losses in MONAI's generative module.
lpips
# Common Python libraries
pandas
numpy
nibabel
tqdm
tensorboard
matplotlib
datasets