USRNet环境搭建和模型测试|简记

Deep Unfolding Network for Image Super-Resolution

代码： https://github.com/cszn/USRNet

论文解析参考知乎

---

环境搭建

服务器：Ubuntu16.04 GTX1060 6G * 8
个人用户：CUDA版本 9.2；cudnn 7.6.2 ; 因此适配的pytorch为 1.4.0+cu92 torchvision==0.5.0+cu92

conda create -n torch14  python=3.6.6
source activate torch14

pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

pip install opencv-python

pip install scipy

pip install matplotlib

使用预训练模型进行测试

nohup python main_test_table1.py &

---

测试代码分析：

 psnr = util.calculate_psnr(img_E, img_H, border=sf**2) 
              

---

---