2D UNet Image Segmentation for HipMRI dataset #269

Open

shans-cloud wants to merge 13 commits into

shakes76:topic-recognitionfrom

shans-cloud:topic-recognition

Commits on Oct 24, 2025

Add utils.py for MRI data loading from appendix section.
shans-cloud
committed
Add HipMRIProstateDataset class for loading and preprocessing 2D MRI slices
shans-cloud
committed

Commits on Oct 25, 2025

Add image resizing and visualisation helpers for dataset
shans-cloud
committed

Commits on Oct 28, 2025

Commits on Nov 2, 2025

Refactor HipMRIProstateDataset to support multi-class segmentation and v2 transforms
shans-cloud
committed
Refactor UNET for multi-class segmentation with raw logits output
shans-cloud
committed
Refactor utils.py for multi-class segmentation and visualisation improvements
shans-cloud
committed
Initial commit: finalised report for U-Net prostate segmentation project
shans-cloud
committed
Add finalised train.py implementing U-Net training with Focal and Dice Loss
shans-cloud
committed
Add predict.py for evaluating U-Net on HipMRI test set
shans-cloud
committed
Add final result and visualisation images for report
shans-cloud
committed