Segmenting MRI images using 2D Unet#142

Open

NicMarchant wants to merge 24 commits intoshakes76:topic-recognitionfrom NicMarchant:topic-recognition

+374-29

Commits on Oct 7, 2024

Modified dataset.py to add the dataloader for the three keras_slices datasets
Deb
committed
added the loading of the segmented data to dataset.py as previously it was just the unsegmented data, added import calls to train.py to access each dataset
Deb
committed

Commits on Oct 10, 2024

Added encoder, bottleneck, decoder and unet functions to modules.py.
Deb
committed
Tested the functionality of the model with a subset of data.
Deb
committed
Adding files to rangpur to test model. using 50 epochs and a batch size of 32
Deb
committed

Commits on Oct 11, 2024

Added loss, accuracy and dice score variables to store the results of the training,
Deb
committed

Commits on Oct 16, 2024

Modified train to use Binary cross entropy, this allows it to run on RANGPUR.
Deb
committed

Commits on Oct 21, 2024

Added code to predict.py to show the first 5 original images,
kankuna
committed
Cleaned imports and added dice
kankuna
committed
Modifying training.py for accuracy
kankuna
committed
Sorted data and changed folder
kankuna
committed
Added a learning rate scheduler to help dice score
kankuna
committed

Commits on Oct 22, 2024

Changed to combined loss function better accuracy
kankuna
committed