Skip to content
A simple implementation of 3D-Unet on a 3D Prostate Segmentation Task
Jupyter Notebook
Branch: master
Clone or download
Latest commit c579e18 Mar 1, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore Initial commit Mar 1, 2018
3D_unet.ipynb Unet Implementation Mar 1, 2018
README.md Update README.md Mar 1, 2018

README.md

3D-Unet

A single-file implementation of 3D unet on a 3D Prostate Segmentation Task

A reimplementation of: https://arxiv.org/pdf/1606.06650.pdf which is based off of https://arxiv.org/pdf/1505.04597.pdf

The dataset was acquired from here: https://wiki.cancerimagingarchive.net/display/Public/NCI-ISBI+2013+Challenge+-+Automated+Segmentation+of+Prostate+Structures

Results aren't too great but were trained for only 10000 iterations using an old laptop. If you get the chance to rerun the code, do let me know if you're able to get any better results/if anything is wrong!

You can’t perform that action at this time.