ImageSegment with Unet3D
This is an example of 3D ImageSegment
Prerequisities
The following dependencies are needed:
- numpy >= 1.11.1
- SimpleITK >=1.0.1
- opencv-python >=3.3.0
- tensorflow-gpu ==1.8.0
- pandas >=0.20.1
- scikit-learn >= 0.17.1
How to Use
(re)implemented the model with tensorflow in the paper of "Çiçek, Ö., Abdulkadir, A., Lienkamp, S. S., Brox, T., and Ronneberger, O. (2016) 3D U-net: Learning dense volumetric segmentation from sparse annotation. MICCAI 2016"
Unet3d structure is in model create_conv_net function
My Machine is GTX1080,when train it,the batchsize is one,too many batchsize lead to memory out.If you want sovel the bottleneck,you can change GTX1080 to GTX1080Ti,or take mutilGPU for training
Contact
- https://github.com/junqiangchen
- email: [email protected]
- WeChat Public number: 最新医学影像技术