search

ZFTurbo / ZF_UNET_224_Pretrained_Model

Modification of convolutional neural net "UNET" for image segmentation in Keras framework
GNU General Public License v3.0
214 stars 64 forks source link

ZF architecture #16

Open AmericaBG opened 4 years ago

AmericaBG commented 4 years ago

Hi! Could you tell me what changes in UNET to become ZF unet? Thank you very much!

ZFTurbo commented 4 years ago

As I remember: different number of convolutions and filters in each block. Also SpatialDropout before final layer. But overall it's very close to original UNET.