Imported from https://github.com/AdarshMJ/Underwater-Image-Enhancement-via-Style-Transfer