Add inference code and fix a parameter bug in u-net

TaoHuang2018 / Neighbor2Neighbor

Neighbor2Neighbor: Self-Supervised Denoising from Single Noisy Images

BSD 3-Clause "New" or "Revised" License

248 stars 39 forks source link

Closed TomHeaven closed 2 years ago

TomHeaven commented 2 years ago

add inference code to fix https://github.com/TaoHuang2018/Neighbor2Neighbor/issues/9
fix "TypeError: conv_transpose2d(): argument 'output_padding' (position 6) must be tuple of ints, not tuple" in arch_unet.py for Pytorch 1.11+.