iwpodnet_pytorch

Pytorch implementation of IWPOD-Net (https://github.com/claudiojung/iwpod-net).

What is this

• iwpod.py contains pytoch implementation of IWPOD-NET
• keras_to_torch.py contains function that allows loading original keras wieghts into the pytorch model
• save_keras_weigths.py small script serialize and dump keras weights

Important

When using remember that PyTorch uses (B,C,W,H) whereas Keras uses (B,W,H,C) tensor convention.