Imported from https://github.com/itsuki8914/simply-noise2noise-TF