Imported from https://github.com/weichen582/RetinexNet