Topic recognition#174

Open

prianka-indla wants to merge 30 commits intoshakes76:mainfrom

prianka-indla:topic-recognition

Commits on Sep 16, 2024

Added recognition branch for recognition problems to be committed.
shakes76
committed

Commits on Oct 29, 2024

Add modules.py with initial UNet model implementation
Miss Prianka Indla
committed
Add dataset.py with Dataset class initialised that loads Nifti files
Miss Prianka Indla
committed
Add train.py to train and validate model
Miss Prianka Indla
committed
Add predict.py to test the trained model
Miss Prianka Indla
committed
Add test_driver.py which is the SLURM file to run code on GPU
Miss Prianka Indla
committed
Add dataset.py - fixed converting segmentation mask to binary for accurate dice score calculations
Miss Prianka Indla
committed
Add README.md to describe alogirthm workflow.
Miss Prianka Indla
committed
Included dropout functionality
Miss Prianka Indla
committed
Update train.py to ensure input data shape matches expected dimensions
Miss Prianka Indla
committed
Update train.py to calculate dice coefficient
Miss Prianka Indla
committed
Update train.py to adjust number of epochs and learning rate to reach 0.7 min dice coefficient
Miss Prianka Indla
committed
Update dataset.py updated docstrings
Miss Prianka Indla
committed
Update train.py to include early stopping
Miss Prianka Indla
committed
Update train.py to remove early stopping as it was ineffective in improving model efficiency
Miss Prianka Indla
committed
Update modules.py update UNet encoder functionality
Miss Prianka Indla
committed
Update dataset.py to normalise masking
Miss Prianka Indla
committed
Testing dice coefficient min 0.7 with 10 epochs
Miss Prianka Indla
committed
Fixing indentation
Miss Prianka Indla
committed
Add plots and images of results
Miss Prianka Indla
committed
Update README.md
prianka-indla
authored
Update README.md
prianka-indla
authored
Update modules.py to use BCEDiceLoss function for higher efficiency
Miss Prianka Indla
committed
Save all changes 30/08
Miss Prianka Indla
committed
Save all changes 30/08
Miss Prianka Indla
committed
Update predict.py added docstrings
Miss Prianka Indla
committed
Update dice score vs epoch analysis and graph
Miss Prianka Indla
committed
Update README.md
prianka-indla
authored
Update README.md
prianka-indla
authored
Update README.md
prianka-indla
authored

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Topic recognition#174

Topic recognition#174
prianka-indla wants to merge 30 commits intoshakes76:mainfrom
prianka-indla:topic-recognition

Commits on Sep 16, 2024

Added recognition branch for recognition problems to be committed.

Commits on Oct 29, 2024

Add modules.py with initial UNet model implementation

Add dataset.py with Dataset class initialised that loads Nifti files

Add train.py to train and validate model

Add predict.py to test the trained model

Add test_driver.py which is the SLURM file to run code on GPU

Add dataset.py - fixed converting segmentation mask to binary for accurate dice score calculations

Add README.md to describe alogirthm workflow.

Included dropout functionality

Update train.py to ensure input data shape matches expected dimensions

Update train.py to calculate dice coefficient

Update train.py to adjust number of epochs and learning rate to reach 0.7 min dice coefficient

Update dataset.py updated docstrings

Update train.py to include early stopping

Update train.py to remove early stopping as it was ineffective in improving model efficiency

Update modules.py update UNet encoder functionality

Update dataset.py to normalise masking

Testing dice coefficient min 0.7 with 10 epochs

Fixing indentation

Add plots and images of results

Update README.md

Update README.md

Update modules.py to use BCEDiceLoss function for higher efficiency

Save all changes 30/08

Save all changes 30/08

Update predict.py added docstrings

Update dice score vs epoch analysis and graph

Update README.md

Update README.md

Update README.md