Imported from https://github.com/Net-Mist/style-transfer-tf2