3D U-Net for Prostate MRI Segmentation – Janvhi Sharma (s4975045)#279

Open

JS-0911 wants to merge 29 commits intoshakes76:topic-recognitionfrom

JS-0911:topic-recognition

Commits on Nov 1, 2025

chore: add .gitignore to exclude datasets, models, and venv per spec
JS-0911
committed
chore: initialise project structure with dataset, modules, train, predict, and test scripts
JS-0911
committed
feat(data): finalise dataset loader with resize/normalize/augment pipeline (verified locally)
JS-0911
committed
feat(model): verified Improved 3D U-Net architecture (residual blocks + dropout); architecture finalised
JS-0911
committed
feat(train): refined training/validation loops with Dice loss, seeding, and checkpoint management
JS-0911
committed
feat(eval): integrated Dice coefficient evaluation and final testing routine for trained model
JS-0911
committed
fix(import): remove relative imports and make train/predict runnable on Rangpur
s4975045
committed
fix(import): remove relative imports and make train/predict runnable on Rangpur
s4975045
committed

Commits on Nov 2, 2025

fix: update Improved3DUNet imports, dataset + train refactor
s4975045
committed
wip: update slurm or misc tweaks
s4975045
committed
fix: corrected rotation axes in RandomRotate3D transform
s4975045
committed

Commits on Nov 3, 2025

fix: corrected predict.py import and added final SLURM scripts (Dice 0.7976)
s4975045
committed

Commits on Nov 4, 2025

Commits on Nov 25, 2025

Fixed file structure: moved all files to correct student ID folder
JS-0911
committed