Imported from https://github.com/Minerva-jiezhao/PGU-net-Model