Skip to content

3D Unet Convolution Neural Network (CNN) designed for brain tumor segmentation

License

Notifications You must be signed in to change notification settings

mjiansun/3DUnetCNN

 
 

Repository files navigation

3D U-Net Convolution Neural Network with Keras

Designed after this paper on volumetric segmentation with a 3D U-Net. Currently, the network does not have the B-Spline deformations that are mentioned in the paper. If you figure out a way to apply these to a 3D Keras CNN, let me know! PRs are always welcome!

The code was written to be trained using the BRATS data set for brain tumors, but it can be easily modify to be used in other 3D applications. To adapt the network, you might have to play with the input size to get something that works for your data.

I used Bohdan Pavlyshenko's Kaggle kernel for 2D U-Net segmentation as a base for this 3D U-Net.

About

3D Unet Convolution Neural Network (CNN) designed for brain tumor segmentation

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%