源码:https://github.com/zhangyhuaee/KinD
1.打开GIT bash
git clone https://github.com/zhangyhuaee/KinD.git
2.下载预训练的检查点checkpoint,解压后与源代码相比多了一个文件,我选择直接覆盖了原文件中的checkpoint。
3.在刚刚创建的配置了TensorFlow的虚拟环境中,运行代码:
python evaluate.py
出现很多问题:
一:很多库没安装:(skimagea安装报错,因为他本命叫scikit-image)
conda install scipy
conda install scikit-image
二:InternalError:failed to create session
GPU内存不够了,在代码前设置export CUDA_VISIBLE_DEVICE=0。
问题解决后:
查看路径,可以看到是对KinD\test中的三张图片进行增强,结果保存在KinD\result\test中: