search

xmengli / H-DenseUNet

TMI 2018. H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation from CT Volumes
526 stars 161 forks source link

2d denseunet use #55

Open AmericaBG opened 5 years ago

AmericaBG commented 5 years ago

Hi! First of all, thank you for sharing your code!

I have one question, denseunet is a 2d network but when you look at the code..."img_input = Input(shape=(3, 224, 224), name='data')" So, can we really train the network with 2d data?

Thank you!

xmengli commented 5 years ago

Yes, we slice the 3D volumes to 2d slices.