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

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

Primary LanguagePythonMIT LicenseMIT

Stargazers