1相关准备
相应的pytorch环境这里就不赘述了,enlightenGan的源码在github可以直接下载,大家自行下载。
2测试环节
1:用pycharm打开下载好的源码,按下图的目录结构创建文件夹:checkpoints/enlightening和model
2:在checkpoints/enlightening和model文件夹下分别放入网络模型和VGG的模型,模型可以在github上下载到,下载好了拉里面就行。
3:创建测试文件夹test_dataset。这个一定要注意!这个文件夹一定要跟您下载的enlightenGan源码同级!里面在创建文件夹testA,testB.testA里面放入你要恢复的图片。testB里面随意放入图片,但不能不放!
4:打开pycharm的Terminal,输入命令 pip install -r requirement.txt 下载所需的包