Skip to content

Latest commit

 

History

History
32 lines (23 loc) · 1.27 KB

README.md

File metadata and controls

32 lines (23 loc) · 1.27 KB
Notebook Colab Link
Guiding VQGAN for fMRI Reconstruction Open In Colab
StyleGAN Reconstruction Open In Colab
Voxel to CLIP-aligned voxels Open In Colab

Currently we are trying to replicate this preprint: https://arxiv.org/abs/2210.01769

Installation

To create a conda environment that will run the notebooks:

conda env create -f environment.yaml
conda activate medical-v1

The setup.sh script lists the conda and pip commands to create this environment manually. To download the checkpoints and other data:

./download.sh

Train DiffusionPrior

Train the prior model (see the file for all the command line arguments):

python train_prior.py

Results

There are some training runs on wandb: https://wandb.ai/jimgoo/laion-fmri