Skip to content

Latest commit

 

History

History
10 lines (8 loc) · 475 Bytes

README.md

File metadata and controls

10 lines (8 loc) · 475 Bytes
Raw

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.