Skip to content

dllussier/rsfMRI_VAE

Repository files navigation

rsfMRI_VAE

Goals:
Build 2D, 3D and graph convolutional VAE frameworks for use with resting-state fMRI data.

VAEs are based off of the Pytorch VAE example: https://github.com/pytorch/examples/tree/master/vae

Requirements:
torch==1.0.1
torchmed==0.0.1a0
sklearn==0.22.1
nibabel==3.0.0
numpy==1.18.1
nilearn==0.6.1
torchvision==0.0.2
pillow==6.1.0

For graph frameworks (in addition to above):
dgl==0.4.2
networkx==2.4

About

#work in progress

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages