Closed ravi9 closed 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
libs/pconv_layer.py
model_pconv.py
K.layers.Conv2D
PCONV2D
--use_pconv
settings.py
train.py
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
libs/pconv_layer.py
, this is a keras implementation from the original contribution.model_pconv.py
: This replaces allK.layers.Conv2D
toPCONV2D
.--use_pconv
, this will load the model defined inmodel_pconv.py
which uses layer 'libs/pconv_layer.py`settings.py
file or pass the argument--use_pconv
to thetrain.py
script