相应的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 下载所需的包
5:这个非常重要!!!!!!找到 predict.py中的以下代码,在以下代码之前加入主函数,要不然预测一定会报错!
6:Terminal输入命令:python scripts/script.py --predict。进行预测。然后程序就开始运行了!
ps:在输入 pip install -r requirement.txt python scripts/script.py --predict命令时,可能会说找不到文件,要检查一下你自己的路径,实在不行用绝对路径!
3 结果预测结束后会生成对于的结果图:



