ZFTurbo / Keras-inference-time-optimizer

Optimize layers structure of Keras model to reduce computation time
MIT License
157 stars 18 forks source link

Support for TransposeConv+ Batchnorm #19

Open anilsathyan7 opened 5 years ago

anilsathyan7 commented 5 years ago

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

ZFTurbo commented 5 years ago

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

anilsathyan7 commented 5 years ago
  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
ZFTurbo commented 5 years ago

Thanks, I will look into it.

ZFTurbo commented 5 years ago

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