search

jakeret / tf_unet

Generic U-Net Tensorflow implementation for image segmentation
GNU General Public License v3.0
1.9k stars 748 forks source link

Unet 3D #158

Open MarvinYeh opened 6 years ago

MarvinYeh commented 6 years ago

Hi,

I've updated the Unet3D and also create a demo on jupyter notebook. I don't have a proper 3d dataset to train on so currently it's trained using the generator, hence each slice of image is independent. it's still trainable but the result is not as good as using 2d Unet.

jakeret commented 6 years ago

Thanks for the contribution. I'm wondering if your implementation should not rather go into a package for its own. What do you think?

MarvinYeh commented 6 years ago

Hi, I wasn't quite sure if I understand you correctly. Are you suggesting all the 3D-related features should be within an independent set of functions(or scripts?) from your own and leave all your original functions as it was?

emergencyd commented 6 years ago

Is there any official 3D unet code on tensorflow?