intel / unet

U-Net Biomedical Image Segmentation
Apache License 2.0
303 stars 124 forks source link

adding PCONV support (--use_pconv) #12

Closed ravi9 closed 4 years ago

ravi9 commented 4 years ago

Adding support for Partial Convolution based padding. This is an implementation adapted from https://github.com/MathiasGruber/PConv-Keras This layer is Keras implementation of Partial Convolution based padding. See original contribution here: https://github.com/NVIDIA/partialconv