Imported from https://github.com/FelixGruen/tensorflow-u-net