Non-local-U-Nets-2D-block:非本地U-Net的Inplement 2D全局聚合块。 写在火炬

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