This repository contains all source code necessary to replicate our recent work entitled "Predicting the retinotopic organization of human visual cortex from anatomy using geometric deep learning" available in NeuroImage.
Models were generated using Pytorch Geometric. Since this package is under constant updates, we highly recommend that you follow the following steps to run our models locally:
- Create a conda environment (or docker container)
- Install torch first:
$ pip install torch==0.4.1
$ pip install torchvision==0.2.1
- Install torch-scatter, torch-sparse, torch-cluster, torch-spline-conv and torch-geometric:
$ pip install torch-scatter==1.0.4
$ pip install torch-sparse==0.2.2
$ pip install torch-cluster==1.1.5
$ pip install torch-spline-conv==1.0.4
$ pip install torch-geometric==0.3.1
- Install the remaining required packages that are available at requirements.txt:
$ pip install -r requirements.txt
- Clone DeepRetinotopy:
$ git clone git@github.com:Puckett-Lab/deepRetinotopy.git
Finally, install the following git repository for plots:
$ pip install git+https://github.com/felenitaribeiro/nilearn.git
This folder contains all source code necessary to reproduce all figures and summary statistics in our manuscript.
This folder contains all source code necessary to train a new model and to generate predictions on the test dataset using our pre-trained models.
This folder contains all source code necessary to replicate datasets generation, in addition to functions and labels used for figures and models' evaluation.
Please cite our paper if you used our model or if it was somewhat helpful for you 😉
@article{Ribeiro2021,
author = {Ribeiro, Fernanda L and Bollmann, Steffen and Puckett, Alexander M},
doi = {https://doi.org/10.1016/j.neuroimage.2021.118624},
issn = {1053-8119},
journal = {NeuroImage},
keywords = {cortical surface, high-resolution fMRI, machine learning, manifold, visual hierarchy,Vision},
pages = {118624},
title = {{Predicting the retinotopic organization of human visual cortex from anatomy using geometric deep learning}},
url = {https://www.sciencedirect.com/science/article/pii/S1053811921008971},
year = {2021}
}
Fernanda Ribeiro <fernanda.ribeiro@uq.edu.au>
Alex Puckett <a.puckett@uq.edu.au>