schuurs/Pytorch_CNN-RNN: Ali's CNN+RNN code rewritten from tensorflow into pytorch @ ad21d569c07e9f8576ad47b33f552e5da6d5c3fe

Ali's CNN+RNN code rewritten from tensorflow into pytorch

4 Větve

Ruben Aguilo Schuurs ad21d569c0 Small update for Andrej use		před 11 měsíci
PET_volumes_customtemplate_float32	548d466ee7 clinical data version finished	před 1 rokem
figures	f140373ea3 First, unsuccessful attempt at GuidedGradCam. Trained new model & removed clinical data temporarily	před 1 rokem
original_model	f140373ea3 First, unsuccessful attempt at GuidedGradCam. Trained new model & removed clinical data temporarily	před 1 rokem
utils	f140373ea3 First, unsuccessful attempt at GuidedGradCam. Trained new model & removed clinical data temporarily	před 1 rokem
.gitignore	99ebbe9235 Added gitignore	před 1 rokem
LP_ADNIMERGE.csv	1335d846e1 Initial commit, added all files and initial progress	před 1 rokem
README	ad21d569c0 Small update for Andrej use	před 11 měsíci
ROC_2024-08-13_14.36.png	f140373ea3 First, unsuccessful attempt at GuidedGradCam. Trained new model & removed clinical data temporarily	před 1 rokem
TALOS_main.py	12ba1c5f54 New runs, failed TALOS attempt, preparing remote work for data augmentation and normalization	před 1 rokem
avgloss_epoch_curve_2024-08-13_14.36.png	f140373ea3 First, unsuccessful attempt at GuidedGradCam. Trained new model & removed clinical data temporarily	před 1 rokem
cnn_net.pth	f140373ea3 First, unsuccessful attempt at GuidedGradCam. Trained new model & removed clinical data temporarily	před 1 rokem
cnn_net_data_2024-08-13_14.36.csv	f140373ea3 First, unsuccessful attempt at GuidedGradCam. Trained new model & removed clinical data temporarily	před 1 rokem
confusion_matrix_2024-08-13_14.36.png	f140373ea3 First, unsuccessful attempt at GuidedGradCam. Trained new model & removed clinical data temporarily	před 1 rokem
main.py	ad21d569c0 Small update for Andrej use	před 11 měsíci

		
			
			
				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!!
			
		