The implement of NonLocalUNet is quiet different from the official implement

[Lloyd-Pottiger]

The implement of NonLocalUNet in this repo is VGG encoder + NonLocal center + UNet decoder, but the official implement is: replacing upsample and downsample with Global Aggregation Block, reduce the layers of UNet and replace the Concatenation with Add.
Refer to https://github.com/divelab/Non-local-U-Nets、https://arxiv.org/abs/1812.04103

[JiYuanFeng]

Hi, thank you for pointing this out. In fact, in the paper, we did not compare with NonLocalUNet, instead of comparing with the method of UNet + NonLocal operation.

[Lloyd-Pottiger]

You are right. Maybe you should change NonLocalUNet into UNet+NonLocal in README which may be missleading.

[JiYuanFeng]

Yes, thank you for the suggestion!