栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

EnlightenGan源码测试

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

EnlightenGan源码测试

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  下载所需的包

5:这个非常重要!!!!!!找到 predict.py中的以下代码,在以下代码之前加入主函数,要不然预测一定会报错!

6:Terminal输入命令:python scripts/script.py --predict。进行预测。然后程序就开始运行了!

ps:在输入 pip install -r requirement.txt  python scripts/script.py --predict命令时,可能会说找不到文件,要检查一下你自己的路径,实在不行用绝对路径!

3 结果

预测结束后会生成对于的结果图:

 

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/757400.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号