Keras-inference-time-optimizer icon indicating copy to clipboard operation
Keras-inference-time-optimizer copied to clipboard

Published 20 hours ago • verified publisher icon ZFTurbo

Support for TransposeConv+ Batchnorm

Open anilsathyan7 opened this issue 5 years ago • 4 comments

It seems TransposeConv + BN is not currently supported !!! Any plans to support this combination?? (Anyway this doesn't seem to be difficult to implement ...)

anilsathyan7 avatar Aug 26 '19 06:08 anilsathyan7

Can you point on TransposeConv layer in Keras code or documentation?

ZFTurbo avatar Aug 26 '19 06:08 ZFTurbo

  1. https://github.com/keras-team/keras/blob/master/keras/layers/convolutional.py#L628
  2. https://keras.io/layers/convolutional/
  3. https://github.com/tensorflow/examples/blob/master/tensorflow_examples/models/pix2pix/pix2pix.py#L220
  4. https://github.com/jocicmarko/ultrasound-nerve-segmentation/blob/master/train.py

anilsathyan7 avatar Aug 26 '19 08:08 anilsathyan7

Thanks, I will look into it.

ZFTurbo avatar Aug 26 '19 08:08 ZFTurbo

I added support for Conv2DTranspose layer. Currently it available in repo. I'll make pypi release a little bit later,

ZFTurbo avatar Aug 31 '19 19:08 ZFTurbo