Non-local-U-Nets-2D-block
Non-local-U-Nets-2D-block copied to clipboard
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.
尝试了一下,非常占显存,论文中作者通过把3D图像切成小块处理的
Origin tensorflow codes
https://github.com/divelab/Non-local-U-Nets