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

Open

JS-0911 wants to merge 29 commits into

shakes76:topic-recognitionfrom

JS-0911:topic-recognition

Commits on Oct 28, 2025

init(project7): created prostate3d_unet3d_jsharma folder and file structure
JS-0911
committed
86c2f9b
feat(dataset): added 3D NIfTI loader with normalization, resize & one-hot masks
JS-0911
committed
d86772b
feat(model): implemented Improved UNet3D with residual conv blocks, BN, and DiceLoss
JS-0911
committed
28c5016
feat(train): added full UNet3D training loop with Dice metric and checkpoint saving
JS-0911
committed
bebbcb4
feat(predict): added inference + Dice evaluation and visualization for 3D UNet
JS-0911
committed
ace008a

Commits on Nov 1, 2025

Commits on Nov 2, 2025

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

Commits on Nov 3, 2025

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

Commits on Nov 4, 2025

README.md First draft
JS-0911
authored
066160d
Update README.md
JS-0911
authored
c9a33b1

Commits on Nov 5, 2025

Update README.md 2
JS-0911
authored
ef367d5
Delete train_job_debug.slurm
JS-0911
authored
af3baae
Delete train_job_final.slurm
JS-0911
authored
aab983a
Update train_output.txt
JS-0911
authored
a3d5581
Delete train_error.txt
JS-0911
authored
5c85e6a
Delete logs directory
JS-0911
authored
30733bc
Delete recognition/prostate3d_unet3d_jsharma/README.md
JS-0911
authored
d6f8db6
final training and segmentation visualisations
JS-0911
authored
c4eb6aa
Update README.md final
JS-0911
authored
25f30ad

Commits on Nov 25, 2025

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