参考链接：

1. EnlightenGAN 开源代码运行
2. EnlightenGAN的运行环境搭建和训练自己的数据

源码下载和环境配置比较简单，本文测试环境：Win10 + RTX3060、cuda 11.3、python 3.8 + torch 1.12.0 + numpy 1.20.1

如果想修改在项目里创建test_daatset文件
(1) 将script.py文件中， ../final_dataset 更改为 final_dataset，然后在自己根目录下（其实就是文件夹EnlightenGAN中）创建一个文件夹 final_dataset，然后把数据trainA和trainB放入到里面，就是文中提到的real_A和real_B。（具体数据和模型，原作者github下有链接）在根目录下建一个model文件夹，把vgg的模型放到里面就很好了。
（2）script.py文件中将–gpu_ids 0,1,2 更改为 --gpu_ids 0，如果你是一个GPU就是这么改，要不然报错。

可能报错cuda内存不够，可以少几张照片运行：