Ali's CNN+RNN code rewritten from tensorflow into pytorch
Ruben Aguilo Schuurs
ad21d569c0
Small update for Andrej use
|
3 months ago | |
|---|---|---|
| PET_volumes_customtemplate_float32 | 10 months ago | |
| figures | 8 months ago | |
| original_model | 8 months ago | |
| utils | 8 months ago | |
| .gitignore | 1 year ago | |
| LP_ADNIMERGE.csv | 1 year ago | |
| README | 3 months ago | |
| ROC_2024-08-13_14.36.png | 8 months ago | |
| TALOS_main.py | 8 months ago | |
| avgloss_epoch_curve_2024-08-13_14.36.png | 8 months ago | |
| cnn_net.pth | 8 months ago | |
| cnn_net_data_2024-08-13_14.36.csv | 8 months ago | |
| confusion_matrix_2024-08-13_14.36.png | 8 months ago | |
| main.py | 3 months ago |
README
Pytorch CNN model to diagnose Alzheimer's Disease on PET Scans
PyTorch version Ruben Aguilo Schuurs (UW-Madison), reworked from Dr. Alison Deatsch's Tensorflow Version
Files:
- main.py: Controls everything. Calls all of the following files
- utils/CNN.py & utils/CNN_Layers.py: CNN Model structure
- preprocess.py: prepares CustomDatasets training, validation, and testing
- train_methods.py: CNN training code
- Graphs.py: Multiple graphing / results functions
- K-fold.py: Specific graphs
- dataset_sd_mean_finder.py: unused
- show_image.py: show image
- TALOS_main.py: unsuccessful attempt at implementing talos
To run (in main.py):
1. Define the CNN filepath and dataset filepath
2. Change the properties
3. Comment or uncomment the functions (train, load, evaluate, predict) depending on your interest.
-- You don't need to load again after training.
4. See it in action! Training will take a while with large dataset!!
PyTorch version Ruben Aguilo Schuurs (UW-Madison), reworked from Dr. Alison Deatsch's Tensorflow Version
Files:
- main.py: Controls everything. Calls all of the following files
- utils/CNN.py & utils/CNN_Layers.py: CNN Model structure
- preprocess.py: prepares CustomDatasets training, validation, and testing
- train_methods.py: CNN training code
- Graphs.py: Multiple graphing / results functions
- K-fold.py: Specific graphs
- dataset_sd_mean_finder.py: unused
- show_image.py: show image
- TALOS_main.py: unsuccessful attempt at implementing talos
To run (in main.py):
1. Define the CNN filepath and dataset filepath
2. Change the properties
3. Comment or uncomment the functions (train, load, evaluate, predict) depending on your interest.
-- You don't need to load again after training.
4. See it in action! Training will take a while with large dataset!!