model.py : contains all the models tried out, for now NetworkA uses 3D CNN and NetworkB is using Vision Transformer from MONAI (ViT)
TrainModelCNN.py : contains the train and test functions
main.py : contains the functions able to load the data, the setting of hyperparameters and the plot function for train and test loss
Normalizing the data was done in the Skull_stripped notebook by using the method implemented by https://github.com/jcreinhold/intensity-normalization
bax@MacBook-Pro-de-Lucas-3 Figures % scp -r [email protected]:/home/lbackes/mri_data/Figures ./
bax@MacBook-Pro-de-Lucas-3 Thesis % scp -r main.py [email protected]:/home/lbackes/mri_data/
Here are the last two plots given by the alan cluster after 200 and 500 epochs respectively
