weilaweila / Non-local-U-Nets-2D-block

Inplement 2D global aggregation block of Non-local-U-Nets. Written in pytorch.

Non-local U-Net for Biomedical Image Segmentation

Implement 2D and 3D global aggregation block of Non local U-Nets in pytorch.

尝试了一下，非常占显存。

修正一个错误：并不能任意形状，忽视了求导的问题。

Origin tensorflow codes

https://github.com/divelab/Non-local-U-Nets